Searched defs:base_shape (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/coreml/builders/ |
| D | util_test.cc | 71 std::vector<int> base_shape = {2, 2, 3, 4}; in TEST_F() local
83 std::vector<int> base_shape = {1, 2, 3, 4}; in TEST_F() local
93 std::vector<int> base_shape = {2, 2, 3, 4}; in TEST_F() local
101 std::vector<int> base_shape = {2, 2, 3, 4}; in TEST_F() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | mul_test.cc | 178 const std::vector<int> base_shape = {2, 3, 1, 2}; in TEST() local
479 const std::vector<int> base_shape = {2, 3, 1, 2}; in QuantizedWithMixedBroadcast() local
|
| D | add_test.cc | 197 const std::vector<int> base_shape = {2, 3, 1, 2}; in TEST() local
490 const std::vector<int> base_shape = {2, 3, 1, 2}; in QuantizedWithMixedBroadcast() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/spmd/ |
| D | spmd_partitioner.h | 333 PartitionedHlo(HloInstruction* hlo, Shape base_shape, PartitioningState state) in PartitionedHlo()
376 const Shape& base_shape() const { return base_shape_; } in base_shape() function
|
| D | spmd_partitioner_util.cc | 201 Shape GetPaddedShapeForUnevenPartitioning(const Shape& base_shape, in GetPaddedShapeForUnevenPartitioning()
324 HloInstruction* hlo, const Shape& base_shape, in TileToPartialReplicateHaloExchange()
418 HloInstruction* hlo, const Shape& base_shape, in PadFromPartialReplicateShape()
885 HloInstruction* hlo, const Shape& base_shape, in ExchangeHaloAndGetValidData()
|
| D | spmd_partitioner.cc | 1357 auto base_shape = MakePartitionedShape(base_shape_, target); in ReshardToPartialReplicateWithAllGather() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 162 StatusOr<Shape> InferWindowOutputShape(const Shape& base_shape, in InferWindowOutputShape()
1815 Shape base_shape = in InferConvolveShape() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | shape_inference.cc | 1368 const ShapedType& base_shape, const xla::Window& window, in InferWindowOutputShape()
1568 ShapedType base_shape = in InferXlaConvOutputShape() local
|