Searched refs:dims_count (Results 1 – 7 of 7) sorted by relevance
41 const int dims_count = in ProcessBroadcastShapes() local45 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()
55 const int dims_count = input1_shape.DimensionsCount(); in ArgMinMax() local56 for (int i = axis + 1; i < dims_count; ++i) { in ArgMinMax()
150 const int dims_count = 12; in TF_LITE_MICRO_TEST() local152 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() local183 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()
187 const int dims_count = start_indices.size(); in BuildStridedSliceParams() local189 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()
522 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local523 for (int i = 0; i < dims_count; ++i) { in MatchingFlatSize()533 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local534 for (int i = 0; i < dims_count; ++i) { in MatchingFlatSize()545 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local546 for (int i = 0; i < dims_count; ++i) { in MatchingFlatSize()558 const int dims_count = shape.DimensionsCount(); in MatchingFlatSize() local559 for (int i = 0; i < dims_count; ++i) { in MatchingFlatSize()726 const int dims_count = shape.DimensionsCount(); in FlatSizeSkipDim() local727 TFLITE_DCHECK(skip_dim >= 0 && skip_dim < dims_count); in FlatSizeSkipDim()[all …]
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 …]
45 const int32_t dims_count = GetTensorShape(tfT).DimensionsCount(); in ConvertToMliTensorData() local46 *mliT->Rank() = is_bias_tensor ? 1 : dims_count; in ConvertToMliTensorData()49 mliT->Shape()[0] = GetTensorShape(tfT).Dims(dims_count - 1); in ConvertToMliTensorData()52 for (int i = 0; i < dims_count; i++) { in ConvertToMliTensorData()
