Function makeCShape

tensorflow/go/attrs.go:28–34

makeCShape converts a shape specified in C.int64_t into a Shape.

(shape []C.int64_t)

// makeCShape converts a shape specified in C.int64_t into a Shape.
func makeCShape(shape []C.int64_t) Shape {
	s := Shape{dims: make([]int64, len(shape))}
	for i, n := range shape {
		s.dims[i] = int64(n)
	}
	return s
}

// Attr returns the value of an attribute on op. It returns an error if the
// attribute does not exist.