Home
last modified time | relevance | path

Searched defs:output_dims (Results 1 – 25 of 117) sorted by relevance

12345

/external/tensorflow/tensorflow/lite/micro/kernels/
Dresize_nearest_neighbor_test.cc55 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestResizeNearestNeighbor() local
100 const int output_dims[] = {4, 1, 1, 3, 1}; in TF_LITE_MICRO_TEST() local
112 const int output_dims[] = {4, 1, 1, 3, 1}; in TF_LITE_MICRO_TEST() local
124 const int output_dims[] = {4, 1, 1, 3, 1}; in TF_LITE_MICRO_TEST() local
136 const int output_dims[] = {4, 1, 3, 1, 1}; in TF_LITE_MICRO_TEST() local
148 const int output_dims[] = {4, 1, 3, 1, 1}; in TF_LITE_MICRO_TEST() local
160 const int output_dims[] = {4, 1, 3, 1, 1}; in TF_LITE_MICRO_TEST() local
180 const int output_dims[] = {4, 1, 3, 3, 1}; in TF_LITE_MICRO_TEST() local
199 const int output_dims[] = {4, 1, 3, 3, 1}; in TF_LITE_MICRO_TEST() local
218 const int output_dims[] = {4, 1, 3, 3, 1}; in TF_LITE_MICRO_TEST() local
[all …]
Dshape_test.cc51 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestShape() local
75 int output_dims[] = {1, 1}; // this is actually input_shapes shape in TF_LITE_MICRO_TEST() local
86 int output_dims[] = {2, 1, 1}; in TF_LITE_MICRO_TEST() local
97 int output_dims[] = {2, 1, 1}; in TF_LITE_MICRO_TEST() local
108 int output_dims[] = {2, 1, 1}; in TF_LITE_MICRO_TEST() local
119 int output_dims[] = {3, 1, 1, 1}; in TF_LITE_MICRO_TEST() local
130 int output_dims[] = {1, 1}; in TF_LITE_MICRO_TEST() local
Darg_min_max_test.cc56 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestArgMinMaxFloat() local
82 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestArgMinMaxQuantized() local
111 const int output_dims[] = {3, 1, 1, 1}; in TF_LITE_MICRO_TEST() local
125 const int output_dims[] = {3, 1, 1, 1}; in TF_LITE_MICRO_TEST() local
140 const int output_dims[] = {3, 1, 1, 1}; in TF_LITE_MICRO_TEST() local
159 const int output_dims[] = {3, 1, 1, 1}; in TF_LITE_MICRO_TEST() local
178 const int output_dims[] = {3, 1, 1, 1}; in TF_LITE_MICRO_TEST() local
197 const int output_dims[] = {3, 1, 1, 1}; in TF_LITE_MICRO_TEST() local
216 const int output_dims[] = {3, 1, 1, 2}; in TF_LITE_MICRO_TEST() local
235 const int output_dims[] = {3, 1, 1, 2}; in TF_LITE_MICRO_TEST() local
[all …]
Dpad_test.cc99 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestPadFloat() local
126 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestPadV2Float() local
154 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestPadQuantized() local
192 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestPadV2Quantized() local
231 const int output_dims[] = {4, 3, 2, 4, 1}; in TF_LITE_MICRO_TEST() local
245 const int output_dims[] = {4, 3, 3, 3, 3}; in TF_LITE_MICRO_TEST() local
265 const int output_dims[] = {4, 3, 2, 4, 1}; in TF_LITE_MICRO_TEST() local
282 const int output_dims[] = {4, 3, 2, 4, 1}; in TF_LITE_MICRO_TEST() local
307 const int output_dims[] = {4, 3, 2, 4, 1}; in TF_LITE_MICRO_TEST() local
330 const int output_dims[] = {4, 3, 2, 4, 1}; in TF_LITE_MICRO_TEST() local
[all …]
Dreshape_test.cc123 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestReshape() local
144 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestReshapeQuantized() local
167 int output_dims[] = {2, 2, 1}; in TF_LITE_MICRO_TEST() local
183 int output_dims[] = {9, 1, 1, 1, 1, 1, 1, 1, 1, 2}; in TF_LITE_MICRO_TEST() local
199 int output_dims[] = {4, -1, -1, 2, 4}; in TF_LITE_MICRO_TEST() local
219 TfLiteIntArray* output_dims = in TF_LITE_MICRO_TEST() local
241 int output_dims[] = {3, 2, 2, 2}; in TF_LITE_MICRO_TEST() local
273 int output_dims[] = {3, 2, 1, -1}; in TF_LITE_MICRO_TEST() local
305 int output_dims[] = {0}; in TF_LITE_MICRO_TEST() local
339 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TF_LITE_MICRO_TEST() local
Dzeros_like_test.cc30 TfLiteIntArray* output_dims = IntArrayFromInts(input_dims_data); in TestZerosLikeFloat() local
62 TfLiteIntArray* output_dims = IntArrayFromInts(input_dims_data); in TestZerosLikeInt32() local
94 TfLiteIntArray* output_dims = IntArrayFromInts(input_dims_data); in TestZerosLikeInt64() local
Dbatch_to_space_nd_test.cc74 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestBatchToSpaceNdFloat() local
101 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestBatchToSpaceNdQuantized() local
156 const int output_dims[] = {4, 1, 4, 3, 1}; in TF_LITE_MICRO_TEST() local
Dspace_to_batch_nd_test.cc74 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestSpaceToBatchNdFloat() local
101 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestSpaceToBatchNdQuantized() local
156 const int output_dims[] = {4, 1, 4, 3, 1}; in TF_LITE_MICRO_TEST() local
Dactivations_test.cc31 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestReluFloat() local
64 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestRelu6Float() local
100 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestReluUint8() local
141 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestRelu6Uint8() local
182 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestReluInt8() local
222 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestRelu6Int8() local
Dcast_test.cc31 TfLiteIntArray* output_dims = IntArrayFromInts(input_dims_data); in TestCastFloatToInt8() local
63 TfLiteIntArray* output_dims = IntArrayFromInts(input_dims_data); in TestCastInt8ToFloat() local
Dcircular_buffer_test.cc50 const int output_dims[] = {4, 1, num_slots, 1, depth}; in TF_LITE_MICRO_TEST() local
119 const int output_dims[] = {4, 1, num_slots, 1, depth}; in TF_LITE_MICRO_TEST() local
Dtranspose_conv_test.cc108 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestTransposeConvFloat() local
144 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestTransposeConvQuantized() local
232 TfLiteIntArray* output_dims = IntArrayFromInts(tflite::testing::kOutputShape); in TF_LITE_MICRO_TEST() local
266 TfLiteIntArray* output_dims = IntArrayFromInts(tflite::testing::kOutputShape); in TF_LITE_MICRO_TEST() local
Dmaximum_minimum_test.cc34 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestMaxMinFloat() local
75 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestMaxMinQuantized() local
114 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestMaxMinQuantizedInt32() local
Dpack_test.cc57 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestPackTwoInputsFloat() local
90 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestPackThreeInputsFloat() local
124 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestPackTwoInputsQuantized() local
161 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestPackTwoInputsQuantized32() local
Ddequantize_test.cc56 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestDequantizeToFloat() local
79 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestDequantizeToInt32() local
Dlogistic_test.cc75 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestLogisticFloat() local
98 TfLiteIntArray* output_dims = IntArrayFromInts(output_dims_data); in TestLogisticQuantized() local
/external/tensorflow/tensorflow/lite/kernels/internal/reference/
Dlegacy_reference_ops.h50 const Dims<4>& output_dims) { in DepthwiseConv()
76 const Dims<4>& output_dims) { in DepthwiseConv()
90 const Dims<4>& output_dims) { in DepthwiseConv()
105 float* output_data, const Dims<4>& output_dims) { in DepthwiseConv()
121 const Dims<4>& output_dims) { in DepthwiseConv()
155 const Dims<4>& output_dims) { in DepthwiseConv()
174 uint8* output_data, const Dims<4>& output_dims) { in DepthwiseConv()
197 const Dims<4>& output_dims) { in DepthwiseConv()
212 float* output_data, const Dims<4>& output_dims, in Conv()
237 float* output_data, const Dims<4>& output_dims, float* im2col_data, in Conv()
[all …]
Dreduce.h35 const int* output_dims, const int input_num_dims, in Reduce()
94 const int* output_dims, const int input_num_dims, in ReduceSumImpl()
130 const int* output_dims, const int output_num_dims, in ReduceGeneric()
164 const int* output_dims, const int output_num_dims, in Mean()
327 const int* output_dims, in QuantizedMeanOrSum()
/external/tensorflow/tensorflow/compiler/xla/service/
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()
226 std::vector<int64> output_dims, in ScalarIndexedConstantArray()
338 std::vector<int64> output_dims, Shape shape) { in ConstructScalarIndexedArray()
/external/tensorflow/tensorflow/lite/kernels/internal/optimized/
Dlegacy_optimized_ops.h146 const Dims<4>& output_dims) { in DepthwiseConv()
177 const Dims<4>& output_dims) { in DepthwiseConv()
191 const Dims<4>& output_dims) { in DepthwiseConv()
206 float* output_data, const Dims<4>& output_dims) { in DepthwiseConv()
290 const Dims<4>& output_dims) { in DepthwiseConv()
329 const Dims<4>& output_dims) { in DepthwiseConv()
348 uint8* output_data, const Dims<4>& output_dims) { in DepthwiseConv()
371 const Dims<4>& output_dims) { in DepthwiseConv()
690 const Dims<4>& output_dims) { in FullyConnected()
706 float* output_data, const Dims<4>& output_dims) { in FullyConnected()
[all …]
/external/tensorflow/tensorflow/lite/micro/kernels/cmsis_nn/
Dconv.cc77 cmsis_nn_dims output_dims; in Prepare() local
197 cmsis_nn_dims output_dims; in EvalQuantizedPerChannel() local
/external/tensorflow/tensorflow/compiler/tf2xla/kernels/
Done_hot_op.cc39 const int output_dims = indices_dims + 1; in Compile() local
Dtile_ops.cc67 std::vector<int64> output_dims(input_shape.dims()); in Compile() local
/external/tensorflow/tensorflow/compiler/tf2xla/lib/
Dbroadcast.cc32 absl::Span<int64 const> output_dims) { in BroadcastTo()
/external/tensorflow/tensorflow/lite/java/src/test/native/
Dinterpreter_test_jni.cc40 TfLiteIntArray* output_dims = TfLiteIntArrayCopy(input->dims); in Java_org_tensorflow_lite_InterpreterTest_getNativeHandleForDelegate() local

12345