Searched defs:base_shape (Results 1 – 7 of 7) 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/compiler/xla/service/spmd/ |
| D | spmd_partitioner.h | 293 PartitionedHlo(HloInstruction* hlo, Shape base_shape, PartitioningState state) in PartitionedHlo()
324 const Shape& base_shape() const { return base_shape_; } in base_shape() function
|
| D | spmd_partitioner_util.cc | 276 Shape GetPaddedShapeForUnevenPartitioning(const Shape& base_shape, in GetPaddedShapeForUnevenPartitioning()
409 HloInstruction* hlo, const Shape& base_shape, in TileToPartialReplicateHaloExchange()
503 HloInstruction* hlo, const Shape& base_shape, in PadFromPartialReplicateShape()
923 HloInstruction* hlo, const Shape& base_shape, in ExchangeHaloAndGetValidData()
|
| D | spmd_partitioner.cc | 961 auto base_shape = MakePartitionedShape(base_shape_, target); in ReshardToPartialReplicateWithAllGather() local
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | mul_test.cc | 168 const std::vector<int> base_shape = {2, 3, 1, 2}; in TEST() local
424 const std::vector<int> base_shape = {2, 3, 1, 2}; in QuantizedWithMixedBroadcast() local
|
| D | add_test.cc | 182 const std::vector<int> base_shape = {2, 3, 1, 2}; in TEST() local
471 const std::vector<int> base_shape = {2, 3, 1, 2}; in QuantizedWithMixedBroadcast() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | shape_inference.cc | 161 StatusOr<Shape> InferWindowOutputShape(const Shape& base_shape, in InferWindowOutputShape()
1813 Shape base_shape = in InferConvolveShape() local
|