| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | resize_nearest_neighbor_test.cc | 32 const RuntimeShape& output_shape, in TestReferenceResizeNearestNeighbor()
52 RuntimeShape output_shape = {1, 1, 1, 1}; in TEST() local
63 RuntimeShape output_shape = {1, 1, 1, 1}; in TEST() local
75 RuntimeShape output_shape = {1, 1, 1, 1}; in TEST() local
87 RuntimeShape output_shape = {1, 3, 3, 1}; in TEST() local
98 RuntimeShape output_shape = {1, 3, 3, 1}; in TEST() local
109 RuntimeShape output_shape = {1, 3, 3, 1}; in TEST() local
121 RuntimeShape output_shape = {1, 3, 3, 1}; in TEST() local
133 RuntimeShape output_shape = {1, 2, 2, 1}; in TEST() local
144 RuntimeShape output_shape = {1, 2, 2, 1}; in TEST() local
[all …]
|
| D | transpose_utils_test.cc | 25 RuntimeShape output_shape({9}); in TEST() local
43 RuntimeShape output_shape({3, 9}); in TEST() local
63 RuntimeShape output_shape({1, 9}); in TEST() local
82 RuntimeShape output_shape({8, 4, 3}); in TEST() local
104 RuntimeShape output_shape({8, 4, 1}); in TEST() local
126 RuntimeShape output_shape({1, 4, 1}); in TEST() local
146 RuntimeShape output_shape({1, 1, 1}); in TEST() local
166 RuntimeShape output_shape({3, 9, 4, 2}); in TEST() local
190 RuntimeShape output_shape({3, 9, 4, 1}); in TEST() local
213 RuntimeShape output_shape({3, 1, 4, 1}); in TEST() local
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | reshape_test.cc | 40 std::vector<int32_t> output_shape(input_shape.cbegin(), input_shape.cend()); variable
61 std::vector<int32_t> output_shape(input_shape.cbegin(), input_shape.cend()); variable
82 std::vector<int32_t> output_shape(input_shape.cbegin(), input_shape.cend()); variable
103 std::vector<int32_t> output_shape(input_shape.cbegin(), input_shape.cend()); variable
123 std::vector<int32_t> output_shape(input_shape.cbegin(), input_shape.cend()); variable
143 std::vector<int32_t> output_shape(input_shape.cbegin(), input_shape.cend()); variable
214 std::vector<int32_t> output_shape(input_shape.cbegin(), input_shape.cend()); in TEST() local
|
| /external/ComputeLibrary/src/runtime/heuristics/direct_conv/ |
| D | ClDirectConvDefaultConfigBifrost.cpp | 79 …TensorShape output_shape = misc::shape_calculator::compute_deep_convolution_shape(*src, *wei, conv… in configure_G71_f32() local
103 …TensorShape output_shape = misc::shape_calculator::compute_deep_convolution_shape(*src, *wei, conv… in configure_G71_f16() local
127 …TensorShape output_shape = misc::shape_calculator::compute_deep_convolution_shape(*src, *wei, conv… in configure_G71_u8() local
151 …TensorShape output_shape = misc::shape_calculator::compute_deep_convolution_shape(*src, *wei, conv… in configure_default_f32() local
175 …TensorShape output_shape = misc::shape_calculator::compute_deep_convolution_shape(*src, *wei, conv… in configure_default_f16() local
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 96 const RuntimeShape& output_shape, T* output_data) { in Relu()
108 const RuntimeShape& output_shape, T* output_data) { in Relu0To1()
122 const RuntimeShape& output_shape, T* output_data) { in Relu1()
135 const RuntimeShape& output_shape, float* output_data) { in Relu6()
150 const RuntimeShape& output_shape, T* output_data) { in ReluX()
168 const RuntimeShape& output_shape, T* output_data) { in ReluX()
191 const RuntimeShape& output_shape, in BroadcastMulFivefold()
241 const RuntimeShape& output_shape, int16* output_data) { in Mul()
260 const RuntimeShape& output_shape, uint8* output_data) { in Mul()
289 const RuntimeShape& output_shape, int16_t* output_data) { in Sub16()
[all …]
|
| D | sub.h | 37 const RuntimeShape& output_shape, in SubNonBroadcast()
53 const RuntimeShape& output_shape, in SubNonBroadcast()
74 const RuntimeShape& output_shape, in BroadcastSubSlow()
114 const RuntimeShape& output_shape, in BroadcastSubSlow()
154 const RuntimeShape& output_shape, int64_t* output_data) { in BroadcastSubSlow()
191 const RuntimeShape& output_shape, T* output_data) { in BroadcastSubSlow()
230 const RuntimeShape& output_shape, in BroadcastSub16POTSlow()
274 const RuntimeShape& output_shape, T* output_data) { in BroadcastQuantSubSlow()
356 const RuntimeShape& output_shape, uint8_t* output_data) { in Sub()
372 const RuntimeShape& output_shape, int8_t* output_data) { in Sub()
[all …]
|
| D | pad.h | 40 const P* pad_value_ptr, const RuntimeShape& output_shape, in PadImpl()
120 const P* pad_value_ptr, const RuntimeShape& output_shape, in Pad()
130 const int32_t* pad_value_ptr, const RuntimeShape& output_shape, in Pad()
141 const int32_t* pad_value_ptr, const RuntimeShape& output_shape, in Pad()
151 const RuntimeShape& output_shape, T* output_data) { in PadImageStyle()
160 const RuntimeShape& output_shape, in PadImageStyle()
|
| D | logistic.h | 31 const RuntimeShape& output_shape, float* output_data) { in Logistic()
62 const float* input_data, const RuntimeShape& output_shape, in Logistic()
70 const RuntimeShape& output_shape, int16_t* output_data) { in Logistic()
95 const RuntimeShape& output_shape, int8_t* output_data, in Logistic()
|
| D | mul.h | 51 const RuntimeShape& output_shape, T* output_data) { in Mul()
70 const RuntimeShape& output_shape, in Mul()
82 const RuntimeShape& output_shape, uint8_t* output_data) { in Mul()
96 const RuntimeShape& output_shape, in BroadcastMul4DSlow()
146 const RuntimeShape output_shape = in BroadcastMul4DSlow() local
190 const RuntimeShape output_shape = in BroadcastMul4DSlow() local
|
| D | slice.h | 28 const RuntimeShape& output_shape, in Slice()
64 const RuntimeShape& output_shape, T* output_data) { in Slice()
72 const RuntimeShape& output_shape, TfLiteTensor* output) { in Slice()
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | WinogradConvolutionLayerFixture.h | 59 …input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape output_shape, PadStrid… in setup()
109 …input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape output_shape, const Pa… in compute_target()
249 …TensorShape output_shape = compute_winograd_input_transform_shape(TensorInfo(input_shape, 1, data_… in setup() local
297 …TensorType compute_target(TensorShape input_shape, const TensorShape &output_shape, const Winograd… in compute_target()
338 …eTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &output_shape, cons… in compute_reference()
362 …TensorShape output_shape = compute_winograd_filter_transform_shape(TensorInfo(input_shape, 1, dat… in setup() local
411 …TensorType compute_target(TensorShape input_shape, const TensorShape &output_shape, const Winograd… in compute_target()
453 …eTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &output_shape, cons… in compute_reference()
524 …TensorShape output_shape = compute_winograd_output_transform_shape(TensorInfo(input_shape, 1, data… in compute_target() local
569 …TensorShape output_shape = compute_winograd_output_transform_shape(TensorInfo(input_shape,… in compute_reference() local
|
| D | SpaceToBatchFixture.h | 43 …e input_shape, TensorShape block_shape_shape, TensorShape paddings_shape, TensorShape output_shape, in setup()
63 … const TensorShape &block_shape_shape, const TensorShape &paddings_shape, TensorShape output_shape, in compute_target()
116 … const TensorShape &output_shape, DataType data_type, QuantizationInfo quantization_info) in compute_reference()
144 …e input_shape, TensorShape block_shape_shape, TensorShape paddings_shape, TensorShape output_shape, in setup()
156 …e input_shape, TensorShape block_shape_shape, TensorShape paddings_shape, TensorShape output_shape, in setup()
|
| D | CopyFixture.h | 47 void setup(TensorShape input_shape, TensorShape output_shape, DataType data_type) in setup()
61 …TensorType compute_target(const TensorShape &input_shape, const TensorShape &output_shape, DataTyp… in compute_target()
94 …eTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &output_shape, Data… in compute_reference()
|
| D | ReshapeLayerFixture.h | 48 void setup(TensorShape input_shape, TensorShape output_shape, DataType data_type) in setup()
61 …TensorType compute_target(const TensorShape &input_shape, const TensorShape &output_shape, DataTyp… in compute_target()
94 …eTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &output_shape, Data… in compute_reference()
|
| D | DeconvolutionLayerFixture.h | 53 …input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape output_shape, PadStrid… in setup()
153 …ape input_shape, TensorShape weights_shape, const TensorShape bias_shape, TensorShape output_shape, in compute_target()
211 …, const TensorShape &weights_shape, const TensorShape &bias_shape, const TensorShape &output_shape, in compute_reference()
260 …TensorShape output_shape = compute_deconvolution_output_shape(out_dim, input_info, weights… in setup() local
281 …TensorShape output_shape = compute_deconvolution_output_shape(out_dim, input_info, weights… in setup() local
302 …TensorShape output_shape = compute_deconvolution_output_shape(out_dim, input_info, weights… in setup() local
325 …TensorShape output_shape = compute_deconvolution_output_shape(out_dim, input_info, weights… in setup() local
|
| D | PriorBoxLayerFixture.h | 52 …const TensorShape output_shape = misc::shape_calculator::compute_prior_box_shape(input_info, info); in setup() local
60 …TensorType compute_target(TensorShape input_shape, TensorShape output_shape, PriorBoxLayerInfo inf… in compute_target()
95 …eTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &output_shape, Prio… in compute_reference()
|
| D | Im2ColFixture.h | 70 …const TensorShape output_shape = compute_im2col_conv_shape(&input_info, _kernel_dims, _conv_info, … in setup() local
83 …TensorType compute_target(const TensorShape &input_shape, const TensorShape &output_shape, DataTyp… in compute_target()
117 …void compute_reference(const TensorShape &input_shape, const TensorShape &output_shape, DataType d… in compute_reference()
|
| D | DepthToSpaceLayerFixture.h | 43 …void setup(TensorShape input_shape, int32_t block_shape, TensorShape output_shape, DataType data_t… in setup()
59 … TensorType compute_target(TensorShape input_shape, int32_t block_shape, TensorShape output_shape, in compute_target()
96 const TensorShape &output_shape, DataType data_type) in compute_reference()
|
| D | Col2ImFixture.h | 56 …const TensorShape output_shape = compute_col2im_shape(TensorInfo(input_shape, 1, data_type), convo… in setup() local
69 …TensorType compute_target(const TensorShape &input_shape, const TensorShape &output_shape, const S… in compute_target()
103 …eTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &output_shape, unsi… in compute_reference()
|
| D | SpaceToDepthFixture.h | 44 …void setup(TensorShape input_shape, TensorShape output_shape, const int block_shape, DataType data… in setup()
60 …TensorType compute_target(TensorShape input_shape, TensorShape output_shape, const int block_shape, in compute_target()
101 … SimpleTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &output_shape, in compute_reference()
|
| D | DirectConvolutionLayerFixture.h | 70 … const TensorShape output_shape = compute_deep_convolution_shape(input_info, weights_info, info); variable
77 …input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape output_shape, PadStrid… in setup()
149 …hape, TensorShape weights_shape, const TensorShape &bias_shape, TensorShape output_shape, const Pa… in compute_target()
206 …t TensorShape &weights_shape, const TensorShape &bias_shape, const TensorShape &output_shape, cons… in compute_reference()
260 …input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape output_shape, PadStrid… in setup()
273 …input_shape, TensorShape weights_shape, TensorShape bias_shape, TensorShape output_shape, PadStrid… in setup()
|
| D | WeightsReshapeFixture.h | 54 …const TensorShape output_shape = compute_weights_reshaped_shape(TensorInfo(input_shape, 1, data_ty… in setup() local
67 …TensorType compute_target(const TensorShape &input_shape, const TensorShape &output_shape, const b… in compute_target()
118 …eTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &output_shape, cons… in compute_reference()
|
| /external/tensorflow/tensorflow/core/tpu/kernels/ |
| D | image_resize_ops.cc | 43 xla::Shape output_shape = in GetOutputShape() local
55 const xla::Shape& output_shape) { in CompileGrad()
77 auto output_shape = GetOutputShape(ctx); in CompileForward() local
131 auto output_shape = in Compile() local
|
| /external/ComputeLibrary/tests/validation/fixtures/dynamic_fusion/operators/ |
| D | ReshapeFixture.h | 54 void setup(TensorShape input_shape, TensorShape output_shape, DataType data_type) in setup()
67 … TensorType compute_target(TensorShape &input_shape, TensorShape &output_shape, DataType data_type) in compute_target()
119 …eTensor<T> compute_reference(const TensorShape &input_shape, const TensorShape &output_shape, Data… in compute_reference()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | sparse_to_dense.cc | 40 TfLiteStatus Resize(TfLiteContext* context, const TfLiteTensor* output_shape, in Resize()
53 const TfLiteTensor* output_shape, in CheckDimensionsMatch()
129 const TfLiteTensor* output_shape, in ResizeOutputShape()
149 const TfLiteTensor* output_shape; in Prepare() local
205 const TfLiteTensor* output_shape; in SparseToDenseImpl() local
|