Searched refs:Shape (Results 1 - 25 of 412) sorted by relevance

1234567891011>>

/external/tensorflow/tensorflow/compiler/xla/service/
H A Dshape_inference.h16 // Shape inference is used by the XLA service as the user builds up
41 // TODO(b/73352135): Shape inference does not issue very good error messages, in
49 static StatusOr<Shape> InferUnaryOpShape(UnaryOperation operation,
50 const Shape& arg);
51 static StatusOr<Shape> InferUnaryOpShape(HloOpcode opcode,
56 static StatusOr<Shape> InferBinaryOpShape(
57 BinaryOperation operation, const Shape& lhs, const Shape& rhs,
59 static StatusOr<Shape> InferBinaryOpShape(HloOpcode opcode,
65 static StatusOr<Shape> InferTernaryOpShap
[all...]
H A Dshaped_buffer.h43 ShapedBuffer(const Shape& on_host_shape, const Shape& on_device_shape,
49 const Shape& on_host_shape() const { return on_host_shape_; }
53 const Shape& on_device_shape() const { return on_device_shape_; }
98 Shape on_host_shape_;
101 Shape on_device_shape_;
127 ScopedShapedBuffer(const Shape& on_host_shape, const Shape& on_device_shape,
/external/tensorflow/tensorflow/contrib/lite/toco/
H A Dtooling_util_test.cc29 Shape left;
30 Shape right;
37 {Shape({3}), Shape({3}), Agreement::kBroadcast},
38 {Shape({256, 256, 3}), Shape({256, 256, 3}), Agreement::kBroadcast},
39 {Shape({256, 256, 3}), Shape({3}), Agreement::kBroadcast},
40 {Shape({8, 1, 6, 1}), Shape({
[all...]
/external/clang/test/PCH/
H A Denum.h9 enum Shape { enum
16 enum Shape aRoundShape = Circle;
H A Denum.c17 enum Shape s = Triangle;
/external/tensorflow/tensorflow/java/src/test/java/org/tensorflow/
H A DShapeTest.java25 /** Unit tests for {@link Shape}. */
31 assertEquals(-1, Shape.unknown().numDimensions());
32 assertEquals("<unknown>", Shape.unknown().toString());
37 assertEquals(0, Shape.scalar().numDimensions());
38 assertEquals("[]", Shape.scalar().toString());
43 Shape s = Shape.make(2);
48 s = Shape.make(2, 3);
54 s = Shape.make(-1, 2, 3);
84 assertEquals(Shape
[all...]
/external/snakeyaml/src/test/java/org/yaml/snakeyaml/javabeans/
H A DShape.java18 public interface Shape { interface
H A DTriangleBean.java20 private Shape shape;
30 public Shape getShape() {
34 public void setShape(Shape shape) {
H A DTriangle.java18 public class Triangle implements Shape {
/external/tensorflow/tensorflow/compiler/xla/
H A Dshape_util.h155 static int64 ElementsIn(const Shape& shape);
158 static bool HasZeroElements(const Shape& shape);
167 static int64 ByteSizeOf(const Shape& shape, int64 pointer_size = -1);
177 static int64 ByteSizeOfTupleIndexTable(const Shape& shape,
188 static int64 ByteSizeOfElements(const Shape& shape);
193 static int64 ByteSizeOfSparseIndices(const Shape& shape);
197 static string HumanString(const Shape& shape);
198 static string HumanStringWithLayout(const Shape& shape);
207 static StatusOr<Shape> ParseShapeString(tensorflow::StringPiece s);
211 static bool SameDimensions(const Shape
[all...]
H A Dshape_layout.h38 explicit ShapeLayout(const Shape& shape) : shape_(shape) {}
43 tensorflow::Status AssignLayoutToShape(Shape* to_shape) const;
48 bool MatchesLayoutInShape(const Shape& shape) const;
53 tensorflow::Status CopyLayoutFromShape(const Shape& other_shape);
62 const Shape& shape() const { return shape_; }
65 // layout directly on the shape. Shape must not be a tuple.
72 // Resets the layout on the shape to the provided layout. Shape must not be a
84 Shape shape_;
H A Dlayout_util.h44 static Layout GetDefaultLayoutForShape(const Shape& shape);
52 // Sets the default layout on the Shape.
53 static void SetToDefaultLayout(Shape* shape);
57 static Shape GetWithDefaultLayout(const Shape& shape);
64 static tensorflow::Status ValidateLayoutInShape(const Shape& shape);
69 const Shape& shape);
71 // Clears the layout in the given Shape. After this function is called,
73 static void ClearLayout(Shape* shape);
78 // Returns whether the given Shape i
[all...]
H A Dindex_util.h38 const Shape& shape, tensorflow::gtl::ArraySlice<int64> multi_index);
44 const Shape& shape, int64 linear_index);
61 static bool BumpIndices(const Shape& shape,
70 static int64 GetDimensionStride(const Shape& shape, int64 dimension);
74 static bool IndexInBounds(const Shape& shape,
H A Dshape_util.cc72 bool CompareShapes(const Shape& lhs, const Shape& rhs, bool compare_layouts) {
76 [=](const Shape& l, const Shape& r) {
119 StatusOr<Shape> MakeShapeWithLayoutInternal(
130 Shape shape = ShapeUtil::MakeShape(element_type, dimensions);
137 return InvalidArgument("Shape has no layout.");
145 /* static */ bool ShapeUtil::Equal(const Shape& lhs, const Shape& rhs) {
155 /* static */ int64 ShapeUtil::Rank(const Shape
[all...]
/external/tensorflow/tensorflow/compiler/tf2xla/
H A Dshape_util.h27 // Convert an XLA Shape into the equivalent TensorFlow shape. May fail since
29 Status XLAShapeToTensorShape(const xla::Shape& shape,
32 // Convert a TensorShape into the equivalent XLA Shape proto. Unlike Tensorflow,
36 xla::Shape* shape);
/external/gemmlowp/test/
H A Dbenchmark_meta_gemm.cc64 struct Shape { struct
73 Shape(std::int32_t n, std::int32_t m, std::int32_t k) function in struct:Shape
104 double run_gemms(std::vector<Shape>* shapes) {
159 void time_all(std::vector<Shape>* shapes, std::int32_t repetitions,
179 void time_one(Shape* shape, double max_time) {
204 std::vector<Shape> googlenet_gemms;
205 googlenet_gemms.push_back(Shape(12544, 64, 147));
206 googlenet_gemms.push_back(Shape(3136, 64, 64));
207 googlenet_gemms.push_back(Shape(3136, 192, 576));
208 googlenet_gemms.push_back(Shape(78
[all...]
/external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/
H A DShape.java21 public final class Shape { class
23 /** Create a Shape representing an unknown number of dimensions. */
24 public static Shape unknown() {
25 return new Shape(null);
28 /** Create a Shape representing a scalar value. */
29 public static Shape scalar() {
30 return new Shape(new long[0]);
34 * Create a Shape representing an N-dimensional value.
36 * <p>Creates a Shape representing an N-dimensional value (N being at least 1), with the provided
42 * Shape vecto
108 Shape(long[] shape) { method in class:Shape
[all...]
/external/tensorflow/tensorflow/go/
H A Dshape.go24 // Shape represents the (possibly partially known) shape of a tensor that will
27 // The zero-value of a Shape represents a shape with an unknown number of
29 type Shape struct {
33 // ScalarShape returns a Shape representing a scalar.
34 func ScalarShape() Shape {
35 return Shape{dims: make([]int64, 0)}
38 // MakeShape returns a Shape with the provided size of each dimension.
42 func MakeShape(shape ...int64) Shape {
45 return Shape{dims: cpy}
50 func (s Shape) NumDimension
[all...]
/external/tensorflow/tensorflow/compiler/xla/client/
H A Dexecutable_build_options.h39 // set, the service will chose the layout of the result. A Shape is used to
42 ExecutableBuildOptions& set_result_layout(const Shape& shape_with_layout);
43 const Shape* result_layout() const;
66 Shape result_layout_;
H A Dsharding_builder.h45 OpSharding Tile(const Shape& tile_shape, const TileAssignment& tile_assignment);
51 OpSharding Tile1D(const Shape& tile_shape, int64 num_tiles);
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dshape_partition.h67 ShapePartitionAssigner(const Shape& shape) : shape_(shape) {}
77 const Shape& shape_;
84 ShapePartitionIterator(const Shape& shape,
94 const Shape& shape_;
H A Dcpu_layout_assignment.cc80 static Shape RowMajorShape(const Shape& old_shape) {
81 Shape new_shape(old_shape);
88 static Shape ColMajorShape(const Shape& old_shape) {
89 Shape new_shape(old_shape);
113 Shape output_shape(RowMajorShape(convolution->shape()));
114 Shape input_shape(RowMajorShape(lhs_instruction->shape()));
115 Shape filter_shape(RowMajorShape(rhs_instruction->shape()));
136 Shape output_shap
[all...]
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dcudnn_convolution_runner.h74 CudnnConvKind kind, const Shape& input_shape, const Shape& filter_shape,
75 const Shape& output_shape, perftools::gputools::DeviceMemoryBase input_buf,
85 CudnnConvKind kind, const Shape& input_shape, const Shape& filter_shape,
86 const Shape& output_shape, perftools::gputools::DeviceMemoryBase input_buf,
H A Dgemm_thunk.h43 const Shape& lhs_shape, const Shape& rhs_shape,
44 const Shape& output_shape, bool transpose_lhs, bool transpose_rhs,
69 const Shape lhs_shape_;
70 const Shape rhs_shape_;
71 const Shape output_shape_;
/external/tensorflow/tensorflow/compiler/xla/client/lib/
H A Dtesting.cc37 int64 DataSizeOfShape(const Shape& shape) {
43 for (const Shape& s : shape.tuple_shapes()) {
51 ComputationDataHandle BuildFakeDataOpOnDevice(const Shape& shape,
59 for (const Shape& s : shape.tuple_shapes()) {
65 std::unique_ptr<GlobalData> MakeFakeDataViaDeviceOrDie(const Shape& shape,
81 std::unique_ptr<GlobalData> MakeFakeDataOrDie(const Shape& shape,
107 for (const Shape& parameter : program_shape->parameters()) {

Completed in 399 milliseconds

1234567891011>>