/external/tensorflow/tensorflow/compiler/xla/service/ |
H A D | shape_inference.h | 16 // 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 D | shaped_buffer.h | 43 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 D | tooling_util_test.cc | 29 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 D | enum.h | 9 enum Shape { enum 16 enum Shape aRoundShape = Circle;
|
H A D | enum.c | 17 enum Shape s = Triangle;
|
/external/tensorflow/tensorflow/java/src/test/java/org/tensorflow/ |
H A D | ShapeTest.java | 25 /** 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 D | Shape.java | 18 public interface Shape { interface
|
H A D | TriangleBean.java | 20 private Shape shape; 30 public Shape getShape() { 34 public void setShape(Shape shape) {
|
H A D | Triangle.java | 18 public class Triangle implements Shape {
|
/external/tensorflow/tensorflow/compiler/xla/ |
H A D | shape_util.h | 155 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 D | shape_layout.h | 38 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 D | layout_util.h | 44 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 D | index_util.h | 38 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 D | shape_util.cc | 72 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 D | shape_util.h | 27 // 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 D | benchmark_meta_gemm.cc | 64 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 D | Shape.java | 21 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 D | shape.go | 24 // 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 D | executable_build_options.h | 39 // 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 D | sharding_builder.h | 45 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 D | shape_partition.h | 67 ShapePartitionAssigner(const Shape& shape) : shape_(shape) {} 77 const Shape& shape_; 84 ShapePartitionIterator(const Shape& shape, 94 const Shape& shape_;
|
H A D | cpu_layout_assignment.cc | 80 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 D | cudnn_convolution_runner.h | 74 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 D | gemm_thunk.h | 43 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 D | testing.cc | 37 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()) {
|