| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | process_broadcast_shapes.h | 43 const int dims_count = in ProcessBroadcastShapes() local
47 RuntimeShape scalar_shape(dims_count, 1); in ProcessBroadcastShapes()
49 auto extended_shape0 = RuntimeShape::ExtendedShape(dims_count, shape0); in ProcessBroadcastShapes()
50 auto extended_shape1 = RuntimeShape::ExtendedShape(dims_count, shape1); in ProcessBroadcastShapes()
58 for (int i = dims_count - 1; i >= 0; --i) { in ProcessBroadcastShapes()
97 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 | 55 const int dims_count = input1_shape.DimensionsCount(); in ArgMinMax() local
56 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/lite/kernels/internal/optimized/ |
| D | optimized_ops_utils.h | 58 const int dims_count = shape.DimensionsCount(); in MapAsMatrixWithLastDimAsRows() local
59 const int rows = shape.Dims(dims_count - 1); in MapAsMatrixWithLastDimAsRows()
60 const int cols = FlatSizeSkipDim(shape, dims_count - 1); in MapAsMatrixWithLastDimAsRows()
82 const int dims_count = shape.DimensionsCount(); in MapAsArrayWithLastDimAsRows() local
83 const int rows = shape.Dims(dims_count - 1); in MapAsArrayWithLastDimAsRows()
84 const int cols = FlatSizeSkipDim(shape, dims_count - 1); in MapAsArrayWithLastDimAsRows()
|
| D | optimized_ops.h | 275 const int dims_count = weights_shape.DimensionsCount(); in FullyConnected() local
276 const int input_rows = weights_shape.Dims(dims_count - 1); in FullyConnected()
286 lhs_params.cols = weights_shape.Dims(dims_count - 1); in FullyConnected()
287 lhs_params.rows = FlatSizeSkipDim(weights_shape, dims_count - 1); in FullyConnected()
8003 const int dims_count = input1_shape.DimensionsCount(); in ArgMinMax() local
8004 for (int i = axis + 1; i < dims_count; ++i) { in ArgMinMax()
|
| D | legacy_optimized_ops.h | 661 const int dims_count = weights_shape.DimensionsCount(); in FullyConnected() local
662 const int input_rows = weights_shape.Dims(dims_count - 1); in FullyConnected()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 315 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local
316 for (int i = 0; i < dims_count; ++i) { in MatchingFlatSize()
326 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local
327 for (int i = 0; i < dims_count; ++i) { in MatchingFlatSize()
338 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local
339 for (int i = 0; i < dims_count; ++i) { in MatchingFlatSize()
351 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local
352 for (int i = 0; i < dims_count; ++i) { in MatchingFlatSize()
519 const int dims_count = shape.DimensionsCount(); in FlatSizeSkipDim() local
520 TFLITE_DCHECK(skip_dim >= 0 && skip_dim < dims_count); in FlatSizeSkipDim()
[all …]
|
| 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()
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/transforms/ |
| D | dense_to_sparse.cc | 62 const int dims_count = block_size.size(); in PopulateEncodingParams() local
63 traversal_order->resize(dims_count); in PopulateEncodingParams()
64 format->resize(dims_count); in PopulateEncodingParams()
65 for (int i = 0; i < dims_count; i++) { in PopulateEncodingParams()
68 for (int i = 0; i < dims_count - 1; i++) { in PopulateEncodingParams()
71 (*format)[dims_count - 1] = kTfLiteDimSparseCSR; in PopulateEncodingParams()
75 for (int i = 0; i < dims_count; i++) { in PopulateEncodingParams()
77 traversal_order->push_back(block_rank + dims_count); in PopulateEncodingParams()
239 const int dims_count = type.getRank(); in BuildSparsityParameterAttribute() local
240 std::vector<int> shape(dims_count); in BuildSparsityParameterAttribute()
[all …]
|
| /external/tensorflow/tensorflow/lite/delegates/xnnpack/ |
| D | binary_elementwise_tester.cc | 356 int dims_count = Input1Shape().size(); in CreateTfLiteModel() local
358 dims_count); in CreateTfLiteModel()
359 std::vector<int> traversal_order(dims_count); in CreateTfLiteModel()
360 for (int i = 0; i < dims_count; i++) { in CreateTfLiteModel()
393 int dims_count = Input2Shape().size(); in CreateTfLiteModel() local
395 dims_count); in CreateTfLiteModel()
396 std::vector<int> traversal_order(dims_count); in CreateTfLiteModel()
397 for (int i = 0; i < dims_count; i++) { in CreateTfLiteModel()
|
| D | prelu_tester.cc | 201 const int dims_count = SlopeShape().size(); in CreateTfLiteModel() local
203 dims_count); in CreateTfLiteModel()
204 std::vector<int> traversal_order(dims_count); in CreateTfLiteModel()
205 for (int i = 0; i < dims_count; i++) { in CreateTfLiteModel()
|
| D | transpose_conv_tester.cc | 305 const int dims_count = filter_shape.size(); in CreateTfLiteModel() local
307 dims_count); in CreateTfLiteModel()
308 std::vector<int> traversal_order(dims_count); in CreateTfLiteModel()
309 for (int dim = 0; dim < dims_count; dim++) { in CreateTfLiteModel()
|
| D | conv_2d_tester.cc | 308 int dims_count = filter_shape.size(); in CreateTfLiteModel() local
310 dims_count); in CreateTfLiteModel()
311 std::vector<int> traversal_order(dims_count); in CreateTfLiteModel()
312 for (int i = 0; i < dims_count; i++) { in CreateTfLiteModel()
|
| D | depthwise_conv_2d_tester.cc | 317 int dims_count = filter_shape.size(); in CreateTfLiteModel() local
319 dims_count); in CreateTfLiteModel()
320 std::vector<int> traversal_order(dims_count); in CreateTfLiteModel()
321 for (int i = 0; i < dims_count; i++) { in CreateTfLiteModel()
|
| D | xnnpack_delegate.cc | 4411 const int dims_count = NumDimensions(&input_tensor); in VisitTransposeNode() local
4413 std::copy(&perm_data[0], &perm_data[dims_count], perm.begin()); in VisitTransposeNode()
4416 subgraph, dims_count, perm.data(), in VisitTransposeNode()
5035 const int dims_count = NumDimensions(&output_tensor); in PrepareOpsToDelegate() local
5036 std::vector<int> vector_shape(dims_count); in PrepareOpsToDelegate()
5037 for (int i = 0; i < dims_count; i++) { in PrepareOpsToDelegate()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.h | 221 const int dims_count = shape.dimensions_count();
223 for (int i = 0; i < dims_count; i++) {
235 const int dims_count = shape.dimensions_count();
236 std::vector<int> indices(dims_count);
238 for (int i = dims_count - 1; i >= 0; i--) {
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | cl_device.cc | 109 int dims_count = in GetDeviceWorkDimsSizes() local
111 if (dims_count < 3) { in GetDeviceWorkDimsSizes()
114 std::vector<size_t> limits(dims_count); in GetDeviceWorkDimsSizes()
117 sizeof(size_t) * dims_count, limits.data(), nullptr); in GetDeviceWorkDimsSizes()
|
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | test_util.h | 253 const int dims_count = t.traversal_order.size(); in AddConstSparseInput() local
265 dims_count); in AddConstSparseInput()
266 for (int i = 0; i < dims_count; i++) { in AddConstSparseInput()
325 const int dims_count = t.traversal_order.size();
337 dims_count);
338 for (int i = 0; i < dims_count; i++) {
|
| /external/tensorflow/tensorflow/lite/delegates/nnapi/ |
| D | nnapi_delegate.cc | 5286 const int dims_count = output_tensor.dims->size; in DensifyAndDequantizeConstTensor() local
5287 std::vector<int> vector_shape(dims_count); in DensifyAndDequantizeConstTensor()
5288 for (int i = 0; i < dims_count; i++) { in DensifyAndDequantizeConstTensor()
|