| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | resolve_constant_binary.cc | 94 const int dims_count = output_shape.dimensions_count(); in EvaluateBinaryOperatorOnConstantInputs() local
100 ExtendShape(&input0_shape, dims_count); in EvaluateBinaryOperatorOnConstantInputs()
101 ExtendShape(&input1_shape, dims_count); in EvaluateBinaryOperatorOnConstantInputs()
107 CHECK_EQ(input0_shape.dimensions_count(), dims_count); in EvaluateBinaryOperatorOnConstantInputs()
111 std::vector<int> big_sizes(dims_count); in EvaluateBinaryOperatorOnConstantInputs()
112 std::vector<int> small_sizes(dims_count); in EvaluateBinaryOperatorOnConstantInputs()
117 for (int i = 0; i < dims_count; i++) { in EvaluateBinaryOperatorOnConstantInputs()
121 std::vector<int> input0_indices(dims_count); in EvaluateBinaryOperatorOnConstantInputs()
122 std::vector<int> input1_indices(dims_count); in EvaluateBinaryOperatorOnConstantInputs()
123 std::vector<int> modulo_indices(dims_count); in EvaluateBinaryOperatorOnConstantInputs()
[all …]
|
| D | unroll_batch_matmul.cc | 41 const int dims_count = input_array_a.shape().dimensions_count(); in UnrollBatchMatMul3D() local
45 begin_indices_a.resize(dims_count); in UnrollBatchMatMul3D()
67 {-1, input_array_a.shape().dims(dims_count - 1)})}; in UnrollBatchMatMul3D()
77 begin_indices_b.resize(dims_count); in UnrollBatchMatMul3D()
99 {-1, input_array_b.shape().dims(dims_count - 1)})}; in UnrollBatchMatMul3D()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/ |
| D | types.h | 443 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local
444 for (int i = 0; i < dims_count; ++i) { in MatchingFlatSize()
454 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local
455 for (int i = 0; i < dims_count; ++i) { in MatchingFlatSize()
466 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local
467 for (int i = 0; i < dims_count; ++i) { in MatchingFlatSize()
479 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local
480 for (int i = 0; i < dims_count; ++i) { in MatchingFlatSize()
595 const int dims_count = shape.DimensionsCount(); in FlatSizeSkipDim() local
596 TFLITE_DCHECK(skip_dim >= 0 && skip_dim < dims_count); in FlatSizeSkipDim()
[all …]
|
| /external/tensorflow/tensorflow/lite/kernels/internal/ |
| D | types.h | 459 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local
460 for (int i = 0; i < dims_count; ++i) { in MatchingFlatSize()
470 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local
471 for (int i = 0; i < dims_count; ++i) { in MatchingFlatSize()
482 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local
483 for (int i = 0; i < dims_count; ++i) { in MatchingFlatSize()
495 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local
496 for (int i = 0; i < dims_count; ++i) { in MatchingFlatSize()
611 const int dims_count = shape.DimensionsCount(); in FlatSizeSkipDim() local
612 TFLITE_DCHECK(skip_dim >= 0 && skip_dim < dims_count); in FlatSizeSkipDim()
[all …]
|
| D | strided_slice_logic.h | 174 const int dims_count = start_indices.size(); in BuildStridedSliceParams() local
176 op_params.start_indices_count = dims_count; in BuildStridedSliceParams()
177 op_params.stop_indices_count = dims_count; in BuildStridedSliceParams()
178 op_params.strides_count = dims_count; in BuildStridedSliceParams()
179 for (int i = 0; i < dims_count; ++i) { in BuildStridedSliceParams()
|
| /external/tensorflow/tensorflow/lite/toco/ |
| D | tooling_util.h | 223 const int dims_count = shape.dimensions_count();
225 for (int i = 0; i < dims_count; i++) {
237 const int dims_count = shape.dimensions_count();
238 std::vector<int> indices(dims_count);
240 for (int i = dims_count - 1; i >= 0; i--) {
|
| /external/python/cpython3/Python/ |
| D | ast_unparse.c | 736 Py_ssize_t i, dims_count; in append_ast_ext_slice() local
737 dims_count = asdl_seq_LEN(slice->v.ExtSlice.dims); in append_ast_ext_slice()
738 for (i = 0; i < dims_count; i++) { in append_ast_ext_slice()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 45 const int dims_count = in ProcessBroadcastShapes() local
49 RuntimeShape scalar_shape(dims_count, 1); in ProcessBroadcastShapes()
51 auto extended_shape0 = RuntimeShape::ExtendedShape(dims_count, shape0); in ProcessBroadcastShapes()
52 auto extended_shape1 = RuntimeShape::ExtendedShape(dims_count, shape1); in ProcessBroadcastShapes()
60 for (int i = dims_count - 1; i >= 0; --i) { in ProcessBroadcastShapes()
93 int i = dims_count - 1; in ProcessBroadcastShapes()
3887 const int dims_count = input1_shape.DimensionsCount(); in ArgMinMax() local
3888 for (int i = axis + 1; i < dims_count; ++i) { in ArgMinMax()
|
| /external/tensorflow/tensorflow/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 133 const int dims_count = shape.DimensionsCount(); in MapAsMatrixWithLastDimAsRows() local
134 const int rows = shape.Dims(dims_count - 1); in MapAsMatrixWithLastDimAsRows()
135 const int cols = FlatSizeSkipDim(shape, dims_count - 1); in MapAsMatrixWithLastDimAsRows()
157 const int dims_count = shape.DimensionsCount(); in MapAsArrayWithLastDimAsRows() local
158 const int rows = shape.Dims(dims_count - 1); in MapAsArrayWithLastDimAsRows()
159 const int cols = FlatSizeSkipDim(shape, dims_count - 1); in MapAsArrayWithLastDimAsRows()
791 const int dims_count = weights_shape.DimensionsCount(); in FullyConnected() local
792 const int input_rows = weights_shape.Dims(dims_count - 1); in FullyConnected()
|
| /external/tensorflow/tensorflow/contrib/lite/kernels/internal/optimized/ |
| D | optimized_ops.h | 133 const int dims_count = shape.DimensionsCount(); in MapAsMatrixWithLastDimAsRows() local
134 const int rows = shape.Dims(dims_count - 1); in MapAsMatrixWithLastDimAsRows()
135 const int cols = FlatSizeSkipDim(shape, dims_count - 1); in MapAsMatrixWithLastDimAsRows()
190 const int dims_count = shape.DimensionsCount(); in MapAsArrayWithLastDimAsRows() local
191 const int rows = shape.Dims(dims_count - 1); in MapAsArrayWithLastDimAsRows()
192 const int cols = FlatSizeSkipDim(shape, dims_count - 1); in MapAsArrayWithLastDimAsRows()
913 const int dims_count = weights_shape.DimensionsCount(); in FullyConnected() local
914 const int input_rows = weights_shape.Dims(dims_count - 1); in FullyConnected()
|