Home
last modified time | relevance | path

Searched defs:base_shape (Results 1 – 8 of 8) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/coreml/builders/
H A Dutil_test.cc71 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
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dmul_test.cc178 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
H A Dadd_test.cc197 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
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/spmd/
H A Dspmd_partitioner.h333 PartitionedHlo(HloInstruction* hlo, Shape base_shape, PartitioningState state) in PartitionedHlo()
376 const Shape& base_shape() const { return base_shape_; } in base_shape() function
H A Dspmd_partitioner_util.cc201 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()
H A Dspmd_partitioner.cc1357 auto base_shape = MakePartitionedShape(base_shape_, target); in ReshardToPartialReplicateWithAllGather() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A Dshape_inference.cc162 StatusOr<Shape> InferWindowOutputShape(const Shape& base_shape, in InferWindowOutputShape()
1815 Shape base_shape = in InferConvolveShape() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dshape_inference.cc1368 const ShapedType& base_shape, const xla::Window& window, in InferWindowOutputShape()
1568 ShapedType base_shape = in InferXlaConvOutputShape() local