Home
last modified time | relevance | path

Searched refs:axis_idx (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dreduce.cc105 for (int axis_idx = 0; axis_idx < num_axis; ++axis_idx) { in ResizeOutputTensor() local
106 if (axis[axis_idx] == idx || axis[axis_idx] + input_num_dims == idx) { in ResizeOutputTensor()
144 for (int axis_idx = 0; axis_idx < num_axis; ++axis_idx) { in ResizeOutputTensor() local
145 if (axis[axis_idx] == idx || axis[axis_idx] + input_num_dims == idx) { in ResizeOutputTensor()
/external/tensorflow/tensorflow/contrib/lite/kernels/internal/
Dtypes.h342 for (int axis_idx = 0; axis_idx < num_axis; ++axis_idx) { in ReducedOutputOffset() local
343 if (idx == axis[axis_idx]) { in ReducedOutputOffset()
/external/tensorflow/tensorflow/lite/kernels/internal/
Dtypes.h358 for (int axis_idx = 0; axis_idx < num_axis; ++axis_idx) { in ReducedOutputOffset() local
359 if (idx == axis[axis_idx]) { in ReducedOutputOffset()