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