Home
last modified time | relevance | path

Searched refs:output_dims (Results 1 – 25 of 97) sorted by relevance

1234

/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h49 const Dims<4>& output_dims) { in DepthwiseConv() argument
65 bias_data, DimsToShape(output_dims), output_data); in DepthwiseConv()
75 const Dims<4>& output_dims) { in DepthwiseConv() argument
79 output_activation_max, output_data, output_dims); in DepthwiseConv()
89 const Dims<4>& output_dims) { in DepthwiseConv() argument
95 output_data, output_dims); in DepthwiseConv()
104 float* output_data, const Dims<4>& output_dims) { in DepthwiseConv() argument
107 depth_multiplier, output_data, output_dims); in DepthwiseConv()
120 const Dims<4>& output_dims) { in DepthwiseConv() argument
142 bias_data, DimsToShape(output_dims), output_data); in DepthwiseConv()
[all …]
Dreduce.h32 const int* output_dims, const int input_num_dims, in Reduce() argument
88 const int* output_dims, const int input_num_dims, in ReduceSumImpl() argument
96 return Reduce<In, Out>(input_data, input_dims, output_dims, input_num_dims, in ReduceSumImpl()
124 const int* output_dims, const int output_num_dims, in ReduceGeneric() argument
130 if (!InitTensorDataForReduce(output_dims, output_num_dims, init_value, in ReduceGeneric()
142 return Reduce<T, T>(input_data, input_dims, output_dims, input_num_dims, in ReduceGeneric()
153 const int* output_dims, const int output_num_dims, in Mean() argument
160 size_t current = static_cast<size_t>(output_dims[idx]); in Mean()
179 if (!ReduceSumImpl<T, U>(input_data, input_dims, output_dims, input_num_dims, in Mean()
316 const int* output_dims, in QuantizedMeanOrSum() argument
[all …]
/external/tensorflow/tensorflow/lite/micro/kernels/
Darg_min_max_test.cc79 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestArgMinMaxFloat() local
80 const int output_dims_count = ElementCount(*output_dims); in TestArgMinMaxFloat()
88 CreateInt32Tensor(output, output_dims, "output_tensor"), in TestArgMinMaxFloat()
105 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestArgMinMaxQuantized() local
106 const int output_dims_count = ElementCount(*output_dims); in TestArgMinMaxQuantized()
115 CreateInt32Tensor(output, output_dims, "output_tensor"), in TestArgMinMaxQuantized()
134 const int output_dims[] = {3, 1, 1, 1}; in TF_LITE_MICRO_TEST() local
138 axis_values, output_dims, output_data, in TF_LITE_MICRO_TEST()
148 const int output_dims[] = {3, 1, 1, 1}; in TF_LITE_MICRO_TEST() local
152 axis_values, output_dims, output_data, in TF_LITE_MICRO_TEST()
[all …]
Dpad_test.cc118 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestPadFloat() local
119 const int output_dims_count = ElementCount(*output_dims); in TestPadFloat()
126 CreateFloatTensor(output_data, output_dims, "output_tensor")}; in TestPadFloat()
146 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestPadV2Float() local
147 const int output_dims_count = ElementCount(*output_dims); in TestPadV2Float()
155 CreateFloatTensor(output_data, output_dims, "output_tensor")}; in TestPadV2Float()
175 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestPadQuantized() local
176 const int output_dims_count = ElementCount(*output_dims); in TestPadQuantized()
184 CreateQuantizedTensor(output_data, output_dims, output_scale, in TestPadQuantized()
213 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestPadV2Quantized() local
[all …]
Dreshape_test.cc112 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestReshape() local
117 output_data, output_dims, "input_tensor"); in TestReshape()
142 int output_dims[3] = {2, 2, 1}; in TF_LITE_MICRO_TEST() local
147 output_dims, // output_dims in TF_LITE_MICRO_TEST()
156 int output_dims[10] = {9, 1, 1, 1, 1, 1, 1, 1, 1, 2}; in TF_LITE_MICRO_TEST() local
161 output_dims, // output_dims in TF_LITE_MICRO_TEST()
171 int output_dims[5] = {4, -1, -1, 2, 4}; in TF_LITE_MICRO_TEST() local
176 output_dims, // output_dims in TF_LITE_MICRO_TEST()
194 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TF_LITE_MICRO_TEST() local
196 CreateFloatTensor(output_data, output_dims, "input_tensor"); in TF_LITE_MICRO_TEST()
[all …]
Dactivations_test.cc30 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestReluFloat() local
31 const int output_elements_count = ElementCount(*output_dims); in TestReluFloat()
38 CreateFloatTensor(output_data, output_dims, "output_tensor"), in TestReluFloat()
86 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestRelu6Float() local
87 const int output_elements_count = ElementCount(*output_dims); in TestRelu6Float()
94 CreateFloatTensor(output_data, output_dims, "output_tensor"), in TestRelu6Float()
145 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestReluUint8() local
146 const int output_elements_count = ElementCount(*output_dims); in TestReluUint8()
154 CreateQuantizedTensor(output_data, output_dims, output_scale, in TestReluUint8()
210 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestRelu6Uint8() local
[all …]
Dpack_test.cc35 TfLiteIntArray* output_dims = IntArrayFromInitializer(output_dims_data); in TestPackTwoInputsFloat() local
36 const int output_dims_count = ElementCount(*output_dims); in TestPackTwoInputsFloat()
44 CreateFloatTensor(output_data, output_dims, "output_tensor")}; in TestPackTwoInputsFloat()
110 TfLiteIntArray* output_dims = IntArrayFromInitializer(output_dims_data); in TestPackThreeInputsFloat() local
111 const int output_dims_count = ElementCount(*output_dims); in TestPackThreeInputsFloat()
120 CreateFloatTensor(output_data, output_dims, "output_tensor")}; in TestPackThreeInputsFloat()
183 TfLiteIntArray* output_dims = IntArrayFromInitializer(output_dims_data); in TestPackTwoInputsQuantized() local
184 const int output_dims_count = ElementCount(*output_dims); in TestPackTwoInputsQuantized()
194 CreateQuantizedTensor(output_data, output_dims, "output_tensor", 0, 10)}; in TestPackTwoInputsQuantized()
255 TfLiteIntArray* output_dims = IntArrayFromInitializer(output_dims_data); in TestPackTwoInputsQuantized32() local
[all …]
Dmaximum_minimum_test.cc36 TfLiteIntArray* output_dims = IntArrayFromInitializer(output_dims_data); in TestMaxMinFloat() local
37 const int output_dims_count = ElementCount(*output_dims); in TestMaxMinFloat()
45 CreateFloatTensor(output_data, output_dims, "output_tensor"), in TestMaxMinFloat()
92 TfLiteIntArray* output_dims = IntArrayFromInitializer(output_dims_data); in TestMaxMinQuantized() local
93 const int output_dims_count = ElementCount(*output_dims); in TestMaxMinQuantized()
103 CreateQuantizedTensor(output_data, output_dims, "output_tensor", in TestMaxMinQuantized()
149 TfLiteIntArray* output_dims = IntArrayFromInitializer(output_dims_data); in TestMaxMinQuantizedInt32() local
150 const int output_dims_count = ElementCount(*output_dims); in TestMaxMinQuantizedInt32()
160 CreateQuantized32Tensor(output_data, output_dims, "output_tensor", in TestMaxMinQuantizedInt32()
Dsoftmax_test.cc32 TfLiteIntArray* output_dims = IntArrayFromInitializer(output_dims_data); in TestSoftmaxFloat() local
33 const int output_dims_count = ElementCount(*output_dims); in TestSoftmaxFloat()
40 CreateFloatTensor(output_data, output_dims, "output_tensor"), in TestSoftmaxFloat()
96 TfLiteIntArray* output_dims = IntArrayFromInitializer(output_dims_data); in TestSoftmaxQuantized() local
97 const int output_dims_count = ElementCount(*output_dims); in TestSoftmaxQuantized()
105 CreateQuantizedTensor(output_data, output_dims, "output_tensor", in TestSoftmaxQuantized()
162 TfLiteIntArray* output_dims = IntArrayFromInitializer(output_dims_data); in TestSoftmaxQuantizedSigned() local
163 const int output_dims_count = ElementCount(*output_dims); in TestSoftmaxQuantizedSigned()
171 CreateQuantizedTensor(output_data, output_dims, "output_tensor", in TestSoftmaxQuantizedSigned()
Ddepthwise_conv_test.cc108 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestDepthwiseConvFloat() local
109 const int output_dims_count = ElementCount(*output_dims); in TestDepthwiseConvFloat()
118 CreateFloatTensor(output_data, output_dims, "output_tensor"), in TestDepthwiseConvFloat()
137 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestDepthwiseConvQuantizedPerLayer() local
138 const int output_dims_count = ElementCount(*output_dims); in TestDepthwiseConvQuantizedPerLayer()
153 tflite::testing::CreateQuantizedTensor(output_data, output_dims, in TestDepthwiseConvQuantizedPerLayer()
191 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestDepthwiseConvQuantizedPerChannel() local
192 const int output_dims_count = ElementCount(*output_dims); in TestDepthwiseConvQuantizedPerChannel()
212 CreateQuantizedTensor(output_data, output_dims, output_scale, in TestDepthwiseConvQuantizedPerChannel()
541 const int output_dims[] = {4, 1, 1, 2, 4}; in TF_LITE_MICRO_TEST() local
[all …]
Delementwise_test.cc32 TfLiteIntArray* output_dims = IntArrayFromInitializer(output_dims_data); in TestElementwiseFloat() local
33 const int output_dims_count = ElementCount(*output_dims); in TestElementwiseFloat()
40 CreateFloatTensor(output_data, output_dims, "output_tensor")}; in TestElementwiseFloat()
98 TfLiteIntArray* output_dims = IntArrayFromInitializer(output_dims_data); in TestElementwiseBool() local
99 const int output_dims_count = ElementCount(*output_dims); in TestElementwiseBool()
106 CreateBoolTensor(output_data, output_dims, "output_tensor")}; in TestElementwiseBool()
Dconcatenation_test.cc37 TfLiteIntArray* output_dims = IntArrayFromInitializer(output_dims_data); in TestConcatenateTwoInputs() local
45 CreateFloatTensor(output_data, output_dims, "output_tensor")}; in TestConcatenateTwoInputs()
79 const int output_dims_count = ElementCount(*output_dims); in TestConcatenateTwoInputs()
96 TfLiteIntArray* output_dims = IntArrayFromInitializer(output_dims_data); in TestConcatenateQuantizedTwoInputs() local
106 CreateQuantizedTensor(output_data, output_dims, "output_tensor", in TestConcatenateQuantizedTwoInputs()
141 const int output_dims_count = ElementCount(*output_dims); in TestConcatenateQuantizedTwoInputs()
Dprelu_test.cc35 TfLiteIntArray* output_dims = IntArrayFromInitializer(output_dims_data); in TestPreluFloat() local
36 const int output_dims_count = ElementCount(*output_dims); in TestPreluFloat()
43 CreateFloatTensor(output_data, output_dims, "output_tensor"), in TestPreluFloat()
97 TfLiteIntArray* output_dims = IntArrayFromInitializer(output_dims_data); in TestPreluQuantized() local
98 const int output_dims_count = ElementCount(*output_dims); in TestPreluQuantized()
107 CreateQuantizedTensor(output_data, output_dims, "output_tensor", in TestPreluQuantized()
/external/tensorflow/tensorflow/core/kernels/
Deigen_benchmark.h46 Dimensions output_dims(input_dims[0], // batch in SpatialConvolution()
56 static_cast<Scalar*>(device_.allocate(BufferSize(output_dims))); in SpatialConvolution()
63 Output output(output_data, output_dims); in SpatialConvolution()
82 Dimensions output_dims(input_dims[0], // batch in SpatialConvolutionBackwardInput()
94 static_cast<Scalar*>(device_.allocate(BufferSize(output_dims))); in SpatialConvolutionBackwardInput()
99 device_.memset(output_backward_data, 123, BufferSize(output_dims)); in SpatialConvolutionBackwardInput()
102 OutputBackward output_backward(output_backward_data, output_dims); in SpatialConvolutionBackwardInput()
123 Dimensions output_dims(input_dims[0], // batch in SpatialConvolutionBackwardKernel()
135 static_cast<Scalar*>(device_.allocate(BufferSize(output_dims))); in SpatialConvolutionBackwardKernel()
140 device_.memset(output_backward_data, 123, BufferSize(output_dims)); in SpatialConvolutionBackwardKernel()
[all …]
/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Dbroadcast.cc32 absl::Span<int64 const> output_dims) { in BroadcastTo() argument
38 if (input_dims == output_dims) { in BroadcastTo()
42 if (input_dims.size() > output_dims.size()) { in BroadcastTo()
46 absl::StrJoin(output_dims, ","), "]"); in BroadcastTo()
52 for (auto output_it = output_dims.rbegin(); output_it != output_dims.rend(); in BroadcastTo()
59 " to [", absl::StrJoin(output_dims, ","), in BroadcastTo()
88 if (broadcast_shape != output_dims) { in BroadcastTo()
89 output = xla::Reshape(output, output_dims); in BroadcastTo()
/external/XNNPACK/test/
Dbinary-elementwise-operator-tester.h147 std::array<size_t, XNN_MAX_TENSOR_DIMS> output_dims; in TestF32() local
150 std::fill(output_dims.begin(), output_dims.end(), 1); in TestF32()
157 output_dims[i] = std::max(input1_dims[i], input2_dims[i]); in TestF32()
160 std::accumulate(output_dims.begin(), output_dims.end(), size_t(1), std::multiplies<size_t>()); in TestF32()
173 output_stride *= output_dims[i - 1]; in TestF32()
186 for (size_t i = 0; i < output_dims[0]; i++) { in TestF32()
187 for (size_t j = 0; j < output_dims[1]; j++) { in TestF32()
188 for (size_t k = 0; k < output_dims[2]; k++) { in TestF32()
189 for (size_t l = 0; l < output_dims[3]; l++) { in TestF32()
190 for (size_t m = 0; m < output_dims[4]; m++) { in TestF32()
[all …]
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h147 const Dims<4>& output_dims) { in DepthwiseConv() argument
161 const RuntimeShape output_shape = DimsToShape(output_dims); in DepthwiseConv()
178 const Dims<4>& output_dims) { in DepthwiseConv() argument
182 output_activation_max, output_data, output_dims); in DepthwiseConv()
192 const Dims<4>& output_dims) { in DepthwiseConv() argument
198 output_data, output_dims); in DepthwiseConv()
207 float* output_data, const Dims<4>& output_dims) { in DepthwiseConv() argument
210 depth_multiplier, output_data, output_dims); in DepthwiseConv()
291 const Dims<4>& output_dims) { in DepthwiseConv() argument
311 const RuntimeShape output_shape = DimsToShape(output_dims); in DepthwiseConv()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dindexed_array_analysis.cc74 StrJoin(indexed_array->output_dims(), ","), "])"); in ToString()
189 absl::Span<const int64> output_dims, Shape shape) { in FoldGatherOfGather() argument
208 for (int64 gather_dim : source->output_dims()) { in FoldGatherOfGather()
215 for (int64 output_dim : output_dims) { in FoldGatherOfGather()
221 FindIndex(source->output_dims(), source_dim); in FoldGatherOfGather()
222 CHECK_NE(source_dim_for_index_array, source->output_dims().size()); in FoldGatherOfGather()
291 std::vector<int64> output_dims; in ComputeArrayForGather() local
294 output_dims.push_back(i); in ComputeArrayForGather()
299 if (absl::c_linear_search(indexed->output_dims(), source_dim)) { in ComputeArrayForGather()
300 return FoldGatherOfGather(indexed, indices, source_dim, output_dims, in ComputeArrayForGather()
[all …]
Dindexed_array_analysis.h191 absl::Span<const int64> output_dims() const { return output_dims_; } in output_dims() function
195 std::vector<int64> output_dims, Shape shape) in ScalarIndexedArray() argument
199 output_dims_(std::move(output_dims)), in ScalarIndexedArray()
226 std::vector<int64> output_dims, in ScalarIndexedConstantArray() argument
229 std::move(output_dims), std::move(shape)) { in ScalarIndexedConstantArray()
306 absl::Span<const int64> output_dims, Shape shape);
338 std::vector<int64> output_dims, Shape shape) { in ConstructScalarIndexedArray() argument
341 std::move(output_dims), in ConstructScalarIndexedArray()
345 std::move(output_dims), in ConstructScalarIndexedArray()
/external/tensorflow/tensorflow/lite/kernels/
Dexpand_dims.cc44 TfLiteIntArray* output_dims = TfLiteIntArrayCreate(input_dims.size + 1); in ExpandTensorDim() local
45 for (int i = 0; i < output_dims->size; ++i) { in ExpandTensorDim()
47 output_dims->data[i] = input_dims.data[i]; in ExpandTensorDim()
49 output_dims->data[i] = 1; in ExpandTensorDim()
51 output_dims->data[i] = input_dims.data[i - 1]; in ExpandTensorDim()
55 return context->ResizeTensor(context, output, output_dims); in ExpandTensorDim()
Dwhere.cc47 TfLiteIntArray* output_dims = TfLiteIntArrayCreate(2); in ResizeOutputTensor() local
48 output_dims->data[0] = true_count; in ResizeOutputTensor()
49 output_dims->data[1] = cond_rank; in ResizeOutputTensor()
50 return context->ResizeTensor(context, output_tensor, output_dims); in ResizeOutputTensor()
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Dtile_ops.cc67 std::vector<int64> output_dims(input_shape.dims()); in Compile() local
71 "] >= 0, but got ", output_dims[i])); in Compile()
72 output_dims[i] = input_shape.dim_size(i) * multiples[i]; in Compile()
82 auto result = BroadcastTo(ctx->Input("input"), output_dims); in Compile()
/external/tensorflow/tensorflow/compiler/xla/tests/
Dgrouped_convolution_test.cc41 std::vector<int64> output_dims; member
85 config.output_dims = {batch, activation_size + kernel_size - 1, in GetConv2DTestCases()
92 config.output_dims = {batch, output_feature / 32, in GetConv2DTestCases()
96 config.output_dims = {batch, activation_size - kernel_size + 1, in GetConv2DTestCases()
130 absl::StrJoin(spec.output_dims, "x"), "_output_layout_", in GroupedConvolution2DTestDataToString()
163 absl::StrJoin(spec.output_dims, ","), in BuildHloTextGroupedConvolution2D()
189 absl::StrJoin(spec.output_dims, ","), in BuildHloTextGroupedConvolution2D()
214 absl::StrJoin(spec.output_dims, ","), in BuildHloTextGroupedConvolution2D()
Dconv_depthwise_backprop_filter_test.cc38 std::vector<int64> output_dims; member
71 config.output_dims = {output_space_size, output_space_size, feature, 1}; in GetConv2DTestCases()
94 config.output_dims = {output_space_size, output_space_size, feature, 1}; in GetConv2DTestCases()
117 absl::StrJoin(spec.output_dims, "x"), data_type, "_output_layout_", in BatchGroupedConvolution2DTestDataToString()
144 absl::StrJoin(spec.output_dims, ","), in BuildHloTextBatchGroupedConvolution2D()
/external/tensorflow/tensorflow/lite/delegates/gpu/metal/kernels/
Dconcat.cc248 const auto& output_dims = buffers.find(output_id)->second; in ConcatX() local
250 int groups_x = IntegralDivideRoundUp(output_dims.w, groups_size.x); in ConcatX()
251 int groups_y = IntegralDivideRoundUp(output_dims.h, groups_size.y); in ConcatX()
252 int groups_z = IntegralDivideRoundUp(output_dims.c, 4); in ConcatX()
339 const auto& output_dims = buffers.find(output_id)->second; in ConcatY() local
341 int groups_x = IntegralDivideRoundUp(output_dims.w, groups_size.x); in ConcatY()
342 int groups_y = IntegralDivideRoundUp(output_dims.h, groups_size.y); in ConcatY()
343 int groups_z = IntegralDivideRoundUp(output_dims.c, 4); in ConcatY()

1234