Home
last modified time | relevance | path

Searched defs:src_shape (Results 1 – 25 of 48) sorted by relevance

12

/aosp_15_r20/external/ComputeLibrary/tests/validation/fixtures/
H A DCropResizeFixture.h50 …void setup(TensorShape src_shape, TensorShape boxes_shape, Coordinates2D crop_size, InterpolationP… in setup()
70 …TensorType compute_target(const TensorShape &src_shape, const TensorShape &boxes_shape, const Coor… in compute_target()
108 …SimpleTensor<float> compute_reference(const TensorShape &src_shape, const TensorShape &boxes_shape… in compute_reference()
H A DBatchNormalizationLayerFusionFixture.h49 …void setup(TensorShape src_shape, TensorShape w_shape, TensorShape b_shape, TensorShape dst_shape,… in setup()
83 …TensorType compute_target(TensorShape src_shape, TensorShape w_shape, TensorShape b_shape, TensorS… in compute_target()
159 …SimpleTensor<T> compute_reference(TensorShape src_shape, TensorShape w_shape, TensorShape b_shape,… in compute_reference()
H A DFullyConnectedLayerFixture.h353 …void setup(TensorShape src_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape d… in setup()
474 …void setup(TensorShape src_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape d… in setup()
487 …void setup(TensorShape src_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape d… in setup()
H A DReductionOperationFixture.h101 …TensorType compute_target(const TensorShape &src_shape, DataType data_type, unsigned int axis, Red… in compute_target()
130 …SimpleTensor<T> compute_reference(const TensorShape &src_shape, const TensorShape &dst_shape, Data… in compute_reference()
H A DArgMinMaxFixture.h101 …TensorType compute_target(TensorShape &src_shape, DataType data_type, int axis, ReductionOperation… in compute_target()
130 …SimpleTensor<int32_t> compute_reference(TensorShape &src_shape, DataType data_type, int axis, Redu… in compute_reference()
H A DReduceMeanFixture.h84 …TensorType compute_target(TensorShape &src_shape, DataType data_type, Coordinates axis, bool keep_… in compute_target()
114 …SimpleTensor<T> compute_reference(TensorShape &src_shape, DataType data_type, Coordinates axis, bo… in compute_reference()
H A DGEMMReshapeLHSMatrixFixture.h115 TensorShape src_shape = input_shape; in compute_reference() local
H A DPooling3dLayerFixture.h160 void setup(TensorShape src_shape, Pooling3dLayerInfo pool_info, DataType data_type) in setup()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/
H A Dglobal_pooling_to_reduce_op.cc34 bool IsGlobalPooling(const Pooling2DAttributes& attr, const BHWC& src_shape, in IsGlobalPooling()
43 const BHWC& src_shape, const BHWC& dst_shape) { in IsGlobalAveragePooling()
/aosp_15_r20/external/ComputeLibrary/tests/validation/reference/
H A DSliceOperations.cpp52 const TensorShape &src_shape = src.shape(); in slice() local
97 const TensorShape &src_shape = src.shape(); in strided_slice() local
H A DTile.cpp41 const auto src_shape = src.shape(); in tile() local
/aosp_15_r20/external/ComputeLibrary/tests/validation/CL/
H A DDirectConvolutionLayer.cpp100 const auto src_shape = TensorShape(27U, 13U, 2U); in TEST_SUITE() local
143 auto src_shape = TensorShape(33U, 27U, 3U); in TEST_CASE() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
H A Dreduce.cc525 Reduce CreateReduce(const std::set<Axis>& axis_to_reduce, const BHWC& src_shape, in CreateReduce()
533 const BHWDC& src_shape, OperationType op_type, in CreateReduce()
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/
H A Dsimple_selectors.cc194 const BHWC& src_shape, in SelectReduce()
214 const BHWC& src_shape) { in SelectTile()
/aosp_15_r20/external/tensorflow/tensorflow/core/util/
H A Dtensor_format.h592 const TensorShape& src_shape, in ShapeFromFormat()
618 const TensorShape& src_shape, in ShapeFromFilterFormat()
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/
H A Dmkl_swish_op.cc40 TensorShape src_shape = src_tensor.shape(); in Compute_Scalar() local
H A Dmkl_fused_mish_op.cc39 TensorShape src_shape = src_tensor.shape(); in Compute_Scalar() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/
H A Dflops_util.cc32 uint64_t GetConvolutionTransposedFlops(const BHWC& src_shape, in GetConvolutionTransposedFlops()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/
H A Dliteral.cc557 const Shape& src_shape) { in CopyElementsBetween()
1768 auto src_shape = src.subshape(); in CopyElementsWithDynamicBound() local
2663 const auto& src_shape = shape_->tuple_shapes(i); in MutableBorrowingLiteral() local
2722 const auto& src_shape = shape_->tuple_shapes(i); in BorrowingLiteral() local
/aosp_15_r20/external/ComputeLibrary/tests/datasets/
H A DCropResizeDataset.h101 …void add_config(TensorShape src_shape, TensorShape boxes_shape, Coordinates2D crop_size, Interpola… in add_config()
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/ragged/
H A Dragged_resize_image_op_test.py96 def testResizeWithPartialStaticShape(self, src_shape, src_sizes, dst_size): argument
/aosp_15_r20/external/ComputeLibrary/src/gpu/cl/kernels/
H A DClReshapeKernel.cpp82 const cl_int2 src_shape = in configure() local
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/
H A DNETileKernel.cpp102 const auto src_shape = _input->info()->tensor_shape(); in run() local
/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/
H A DCpuReshapeKernel.cpp67 const TensorShape &src_shape = src->info()->tensor_shape(); in reshape_tensor() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dhost_callback_test.cc43 const Shape& src_shape, void* dst_data, size_t dst_size, in ToHostLayout()

12