| /external/tensorflow/tensorflow/lite/micro/kernels/ |
| D | prelu_test.cc | 150 const int dims_count = 12; in TF_LITE_MICRO_TEST() local
152 uint8_t input_quantized[dims_count]; in TF_LITE_MICRO_TEST()
154 uint8_t golden_quantized[dims_count]; in TF_LITE_MICRO_TEST()
157 uint8_t output_data[dims_count]; in TF_LITE_MICRO_TEST()
182 const int dims_count = 12; in TF_LITE_MICRO_TEST() local
183 int8_t input_quantized[dims_count]; in TF_LITE_MICRO_TEST()
185 int8_t golden_quantized[dims_count]; in TF_LITE_MICRO_TEST()
188 int8_t output_data[dims_count]; in TF_LITE_MICRO_TEST()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | process_broadcast_shapes.h | 41 const int dims_count = in ProcessBroadcastShapes() local
45 RuntimeShape scalar_shape(dims_count, 1); in ProcessBroadcastShapes()
47 auto extended_shape0 = RuntimeShape::ExtendedShape(dims_count, shape0); in ProcessBroadcastShapes()
48 auto extended_shape1 = RuntimeShape::ExtendedShape(dims_count, shape1); in ProcessBroadcastShapes()
56 for (int i = dims_count - 1; i >= 0; --i) { in ProcessBroadcastShapes()
95 int i = dims_count - 1; in ProcessBroadcastShapes()
|
| D | densify.h | 33 const int dims_count = output_shape.DimensionsCount(); in Densify() local
34 std::vector<int> vector_shape(dims_count); in Densify()
35 for (int i = 0; i < dims_count; i++) { in Densify()
|
| D | arg_min_max.h | 44 const int dims_count = input1_shape.DimensionsCount(); in ArgMinMax() local
45 for (int i = axis + 1; i < dims_count; ++i) { in ArgMinMax()
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_constant_binary.cc | 95 const int dims_count = output_shape.dimensions_count(); in EvaluateBinaryOperatorOnConstantInputs() local
101 ExtendShape(&input0_shape, dims_count); in EvaluateBinaryOperatorOnConstantInputs()
102 ExtendShape(&input1_shape, dims_count); in EvaluateBinaryOperatorOnConstantInputs()
108 CHECK_EQ(input0_shape.dimensions_count(), dims_count); in EvaluateBinaryOperatorOnConstantInputs()
112 std::vector<int> big_sizes(dims_count); in EvaluateBinaryOperatorOnConstantInputs()
113 std::vector<int> small_sizes(dims_count); in EvaluateBinaryOperatorOnConstantInputs()
118 for (int i = 0; i < dims_count; i++) { in EvaluateBinaryOperatorOnConstantInputs()
122 std::vector<int> input0_indices(dims_count); in EvaluateBinaryOperatorOnConstantInputs()
123 std::vector<int> input1_indices(dims_count); in EvaluateBinaryOperatorOnConstantInputs()
124 std::vector<int> modulo_indices(dims_count); in EvaluateBinaryOperatorOnConstantInputs()
[all …]
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | dense_to_sparse.cc | 54 const int dims_count = block_size.size(); in PopulateEncodingParams() local
55 traversal_order->resize(dims_count); in PopulateEncodingParams()
56 format->resize(dims_count); in PopulateEncodingParams()
57 for (int i = 0; i < dims_count; i++) { in PopulateEncodingParams()
60 for (int i = 0; i < dims_count - 1; i++) { in PopulateEncodingParams()
63 (*format)[dims_count - 1] = kTfLiteDimSparseCSR; in PopulateEncodingParams()
67 for (int i = 0; i < dims_count; i++) { in PopulateEncodingParams()
69 traversal_order->push_back(block_rank + dims_count); in PopulateEncodingParams()
231 const int dims_count = type.getRank(); in BuildSparsityParameterAttribute() local
232 std::vector<int> shape(dims_count); in BuildSparsityParameterAttribute()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | strided_slice_logic.h | 187 const int dims_count = start_indices.size(); in BuildStridedSliceParams() local
189 op_params.start_indices_count = dims_count; in BuildStridedSliceParams()
190 op_params.stop_indices_count = dims_count; in BuildStridedSliceParams()
191 op_params.strides_count = dims_count; in BuildStridedSliceParams()
192 for (int i = 0; i < dims_count; ++i) { in BuildStridedSliceParams()
|
| D | types.h | 522 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local
523 for (int i = 0; i < dims_count; ++i) { in MatchingFlatSize()
533 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local
534 for (int i = 0; i < dims_count; ++i) { in MatchingFlatSize()
545 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local
546 for (int i = 0; i < dims_count; ++i) { in MatchingFlatSize()
558 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local
559 for (int i = 0; i < dims_count; ++i) { in MatchingFlatSize()
674 const int dims_count = shape.DimensionsCount(); in FlatSizeSkipDim() local
675 TFLITE_DCHECK(skip_dim >= 0 && skip_dim < dims_count); in FlatSizeSkipDim()
[all …]
|
| /external/tensorflow/tensorflow/lite/micro/kernels/ceva/ |
| D | types.h | 671 const int dims_count = shape.DimensionsCount();
672 for (int i = 0; i < dims_count; ++i) {
682 const int dims_count = shape.DimensionsCount();
683 for (int i = 0; i < dims_count; ++i) {
694 const int dims_count = shape.DimensionsCount();
695 for (int i = 0; i < dims_count; ++i) {
707 const int dims_count = shape.DimensionsCount();
708 for (int i = 0; i < dims_count; ++i) {
823 const int dims_count = shape.DimensionsCount();
824 TFLITE_DCHECK(skip_dim >= 0 && skip_dim < dims_count);
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | binary_elementwise_tester.cc | 266 int dims_count = Input1Shape().size(); in CreateTfLiteModel() local
268 dims_count); in CreateTfLiteModel()
269 std::vector<int> traversal_order(dims_count); in CreateTfLiteModel()
270 for (int i = 0; i < dims_count; i++) { in CreateTfLiteModel()
292 int dims_count = Input2Shape().size(); in CreateTfLiteModel() local
294 dims_count); in CreateTfLiteModel()
295 std::vector<int> traversal_order(dims_count); in CreateTfLiteModel()
296 for (int i = 0; i < dims_count; i++) { in CreateTfLiteModel()
|
| D | prelu_tester.cc | 148 const int dims_count = SlopeShape().size(); in CreateTfLiteModel() local
150 dims_count); in CreateTfLiteModel()
151 std::vector<int> traversal_order(dims_count); in CreateTfLiteModel()
152 for (int i = 0; i < dims_count; i++) { in CreateTfLiteModel()
|
| D | conv_2d_tester.cc | 256 int dims_count = filter_shape.size(); in CreateTfLiteModel() local
258 dims_count); in CreateTfLiteModel()
259 std::vector<int> traversal_order(dims_count); in CreateTfLiteModel()
260 for (int i = 0; i < dims_count; i++) { in CreateTfLiteModel()
|
| D | depthwise_conv_2d_tester.cc | 258 int dims_count = filter_shape.size(); in CreateTfLiteModel() local
260 dims_count); in CreateTfLiteModel()
261 std::vector<int> traversal_order(dims_count); in CreateTfLiteModel()
262 for (int i = 0; i < dims_count; i++) { in CreateTfLiteModel()
|
| D | xnnpack_delegate.cc | 3173 const int dims_count = output_tensor.dims->size; in PrepareOpsToDelegate() local
3174 std::vector<int> vector_shape(dims_count); in PrepareOpsToDelegate()
3175 for (int i = 0; i < dims_count; i++) { in PrepareOpsToDelegate()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.h | 220 const int dims_count = shape.dimensions_count();
222 for (int i = 0; i < dims_count; i++) {
234 const int dims_count = shape.dimensions_count();
235 std::vector<int> indices(dims_count);
237 for (int i = dims_count - 1; i >= 0; i--) {
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | cl_device.cc | 76 int dims_count = in GetDeviceWorkDimsSizes() local
78 if (dims_count < 3) { in GetDeviceWorkDimsSizes()
81 std::vector<size_t> limits(dims_count); in GetDeviceWorkDimsSizes()
84 sizeof(size_t) * dims_count, limits.data(), nullptr); in GetDeviceWorkDimsSizes()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | test_util.h | 221 const int dims_count = t.traversal_order.size(); in AddConstSparseInput() local
233 dims_count); in AddConstSparseInput()
234 for (int i = 0; i < dims_count; i++) { in AddConstSparseInput()
293 const int dims_count = t.traversal_order.size();
305 dims_count);
306 for (int i = 0; i < dims_count; i++) {
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 150 const int dims_count = shape.DimensionsCount(); in MapAsMatrixWithLastDimAsRows() local
151 const int rows = shape.Dims(dims_count - 1); in MapAsMatrixWithLastDimAsRows()
152 const int cols = FlatSizeSkipDim(shape, dims_count - 1); in MapAsMatrixWithLastDimAsRows()
174 const int dims_count = shape.DimensionsCount(); in MapAsArrayWithLastDimAsRows() local
175 const int rows = shape.Dims(dims_count - 1); in MapAsArrayWithLastDimAsRows()
176 const int cols = FlatSizeSkipDim(shape, dims_count - 1); in MapAsArrayWithLastDimAsRows()
339 const int dims_count = weights_shape.DimensionsCount(); in FullyConnected() local
340 const int input_rows = weights_shape.Dims(dims_count - 1); in FullyConnected()
350 lhs_params.cols = weights_shape.Dims(dims_count - 1); in FullyConnected()
351 lhs_params.rows = FlatSizeSkipDim(weights_shape, dims_count - 1); in FullyConnected()
|
| D | legacy_optimized_ops.h | 660 const int dims_count = weights_shape.DimensionsCount(); in FullyConnected() local
661 const int input_rows = weights_shape.Dims(dims_count - 1); in FullyConnected()
|