| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | CropResizeFixture.h | 51 …void setup(TensorShape src_shape, TensorShape boxes_shape, Coordinates2D crop_size, InterpolationP… in setup()
71 …TensorType compute_target(const TensorShape &src_shape, const TensorShape &boxes_shape, const Coor… in compute_target()
109 …SimpleTensor<float> compute_reference(const TensorShape &src_shape, const TensorShape &boxes_shape… in compute_reference()
|
| D | ReductionOperationFixture.h | 92 …TensorType compute_target(const TensorShape &src_shape, DataType data_type, unsigned int axis, Red… in compute_target()
121 …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()
80 …TensorType compute_target(TensorShape src_shape, TensorShape w_shape, TensorShape b_shape, TensorS… in compute_target()
156 …SimpleTensor<T> compute_reference(TensorShape src_shape, TensorShape w_shape, TensorShape b_shape,… in compute_reference()
|
| D | ArgMinMaxFixture.h | 96 …TensorType compute_target(TensorShape &src_shape, DataType data_type, int axis, ReductionOperation… in compute_target()
125 …SimpleTensor<int32_t> compute_reference(TensorShape &src_shape, DataType data_type, int axis, Redu… in compute_reference()
|
| D | ReduceMeanFixture.h | 75 …TensorType compute_target(TensorShape &src_shape, DataType data_type, Coordinates axis, bool keep_… in compute_target()
105 …SimpleTensor<T> compute_reference(TensorShape &src_shape, DataType data_type, Coordinates axis, bo… in compute_reference()
|
| D | GEMMReshapeLHSMatrixFixture.h | 112 TensorShape src_shape = input_shape; in compute_reference() local
|
| D | PoolingLayerFixture.h | 190 void setup(TensorShape src_shape, PoolingLayerInfo 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/tensorflow/tensorflow/lite/delegates/gpu/cl/kernels/ |
| D | cl_test.cc | 60 auto src_shape = src_cpu[i].shape; in ExecuteGPUOperation() local
114 auto src_shape = src_cpu[i].shape; in ExecuteGPUOperation() local
162 auto src_shape = src_cpu[i].shape; in ExecuteGPUOperation() local
|
| /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/lite/delegates/gpu/metal/kernels/ |
| D | test_util.cc | 71 auto src_shape = src_cpu[i].shape; in ExecuteGPUOperation() local
128 auto src_shape = src_cpu[i].shape; in ExecuteGPUOperation() local
|
| /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/src/core/NEON/kernels/ |
| D | NETileKernel.cpp | 102 const auto src_shape = _input->info()->tensor_shape(); in run() local
|
| /external/ComputeLibrary/tests/validation/NEON/ |
| D | Im2Col.cpp | 144 const TensorShape src_shape = TensorShape(7U, 27U, 13U); in TEST_SUITE() local
|
| D | DepthwiseConvolutionLayerNative.cpp | 118 TensorShape src_shape{ depth, input_size.x(), input_size.y() }; in TEST_SUITE() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/selectors/ |
| D | simple_selectors.cc | 184 const BHWC& src_shape, in SelectReduce()
204 const BHWC& src_shape) { in SelectTile()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | reduce.cc | 475 Reduce CreateReduce(const std::set<Axis>& axis_to_reduce, const BHWC& src_shape, in CreateReduce()
483 const BHWDC& src_shape, OperationType op_type, in CreateReduce()
|
| D | conv_weights_converter_test_util.cc | 35 auto src_shape = in ConvolutionWeightsConverterTest() local
|
| /external/tensorflow/tensorflow/core/util/ |
| D | tensor_format.h | 591 const TensorShape& src_shape, in ShapeFromFormat()
617 const TensorShape& src_shape, in ShapeFromFilterFormat()
|
| /external/ComputeLibrary/examples/ |
| D | neon_cnn.cpp | 68 const TensorShape src_shape(width_src_image, height_src_image, ifm_src_img); in do_setup() local
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal.cc | 453 const Shape& src_shape) { in CopyElementsBetween()
1624 auto src_shape = src.subshape(); in CopyElementsWithDynamicBound() local
2486 const auto& src_shape = shape_->tuple_shapes(i); in MutableBorrowingLiteral() local
2545 const auto& src_shape = shape_->tuple_shapes(i); in BorrowingLiteral() local
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.cc | 458 ShapeHandle src_shape = src_context->output(input_edge->src_output()); in ConstantPartialShape() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_tensor_to_tensor_op.cc | 499 const auto& src_shape = default_value_tensor.shape(); in SetOutput() local
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/metal/ |
| D | inference_context.cc | 99 const auto src_shape = inputs[i]->tensor.shape; in IsGenericAdd() local
|