| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | CropResizeFixture.h | 50 …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()
|
| D | ReductionOperationFixture.h | 101 …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()
|
| D | BatchNormalizationLayerFusionFixture.h | 49 …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()
|
| D | ArgMinMaxFixture.h | 101 …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()
|
| D | ReduceMeanFixture.h | 84 …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()
|
| D | FullyConnectedLayerFixture.h | 353 …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()
|
| D | GEMMReshapeLHSMatrixFixture.h | 115 TensorShape src_shape = input_shape; in compute_reference() local
|
| D | Pooling3dLayerFixture.h | 160 void setup(TensorShape src_shape, Pooling3dLayerInfo pool_info, DataType data_type) in setup()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/ |
| D | global_pooling_to_reduce_op.cc | 34 bool IsGlobalPooling(const Pooling2DAttributes& attr, const BHWC& src_shape, in IsGlobalPooling()
43 const BHWC& src_shape, const BHWC& dst_shape) { in IsGlobalAveragePooling()
|
| /external/ComputeLibrary/tests/validation/reference/ |
| D | SliceOperations.cpp | 52 const TensorShape &src_shape = src.shape(); in slice() local
97 const TensorShape &src_shape = src.shape(); in strided_slice() local
|
| D | Tile.cpp | 41 const auto src_shape = src.shape(); in tile() local
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_swish_op.cc | 40 TensorShape src_shape = src_tensor.shape(); in Compute_Scalar() local
|
| D | mkl_fused_mish_op.cc | 39 TensorShape src_shape = src_tensor.shape(); in Compute_Scalar() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/ |
| D | flops_util.cc | 32 uint64_t GetConvolutionTransposedFlops(const BHWC& src_shape, in GetConvolutionTransposedFlops()
|
| /external/tensorflow/tensorflow/python/ops/ragged/ |
| D | ragged_resize_image_op_test.py | 96 def testResizeWithPartialStaticShape(self, src_shape, src_sizes, dst_size): argument
|
| /external/ComputeLibrary/tests/datasets/ |
| D | CropResizeDataset.h | 101 …void add_config(TensorShape src_shape, TensorShape boxes_shape, Coordinates2D crop_size, Interpola… in add_config()
|
| /external/ComputeLibrary/tests/validation/CL/ |
| D | DirectConvolutionLayer.cpp | 100 const auto src_shape = TensorShape(27U, 13U, 2U); in TEST_SUITE() local
143 auto src_shape = TensorShape(33U, 27U, 3U); in TEST_CASE() local
|
| /external/ComputeLibrary/src/gpu/cl/kernels/ |
| D | ClReshapeKernel.cpp | 82 const cl_int2 src_shape = in configure() local
|
| /external/ComputeLibrary/src/core/NEON/kernels/ |
| D | NETileKernel.cpp | 102 const auto src_shape = _input->info()->tensor_shape(); in run() local
|
| /external/ComputeLibrary/src/cpu/kernels/ |
| D | CpuReshapeKernel.cpp | 67 const TensorShape &src_shape = src->info()->tensor_shape(); in reshape_tensor() local
|
| D | CpuCopyKernel.cpp | 69 const TensorShape src_shape = src->tensor_shape(); in validate_and_configure_window_with_padding() local
|
| /external/tensorflow/tensorflow/compiler/xla/pjrt/ |
| D | host_callback_test.cc | 43 const Shape& src_shape, void* dst_data, size_t dst_size, in ToHostLayout()
|
| /external/ComputeLibrary/tests/validation/NEON/ |
| D | Im2Col.cpp | 149 const TensorShape src_shape = TensorShape(7U, 27U, 13U); in TEST_SUITE() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | reduce.cc | 525 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()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/ |
| D | test_util.cc | 145 auto src_shape = src_cpu[i].shape; in ExecuteGPUOperation() local
|