Home
last modified time | relevance | path

Searched defs:block_shape_shape (Results 1 – 3 of 3) sorted by relevance

/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DSpaceToBatchFixture.h43 …void setup(TensorShape input_shape, TensorShape block_shape_shape, TensorShape paddings_shape, Ten… in setup()
63 …TensorType compute_target(TensorShape input_shape, const TensorShape &block_shape_shape, const Ten… in compute_target()
115 …nsor<T> compute_reference(const TensorShape &input_shape, const TensorShape &block_shape_shape, co… in compute_reference()
144 …void setup(TensorShape input_shape, TensorShape block_shape_shape, TensorShape paddings_shape, Ten… in setup()
156 …void setup(TensorShape input_shape, TensorShape block_shape_shape, TensorShape paddings_shape, Ten… in setup()
H A DBatchToSpaceLayerFixture.h43 …void setup(TensorShape input_shape, TensorShape block_shape_shape, TensorShape output_shape, DataT… in setup()
59 …TensorType compute_target(TensorShape input_shape, TensorShape block_shape_shape, TensorShape outp… in compute_target()
106 …leTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &block_shape_shape, in compute_reference()
/aosp_15_r20/external/tensorflow/tensorflow/core/ops/
H A Darray_ops.cc2267 ShapeHandle block_shape_shape, in SpaceToBatchShapeHelper()
2343 ShapeHandle block_shape_shape, in BatchToSpaceShapeHelper()