Searched refs:flat_index (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/kernels/ |
| D | mirror_pad.cc | 88 int Eval(EvalData<T>* eval_data, int current_dim, int flat_index, in Eval() argument
95 eval_data->output_data[output_index] = eval_data->input_data[flat_index]; in Eval()
99 const int cache_index = current_dim * eval_data->input_size + flat_index; in Eval()
120 output_index = Eval(eval_data, current_dim + 1, flat_index + i * multiplier, in Eval()
125 output_index = Eval(eval_data, current_dim + 1, flat_index + i * multiplier, in Eval()
131 output_index = Eval(eval_data, current_dim + 1, flat_index + i * multiplier, in Eval()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | conv_2d_gpu.h | 154 int flat_index = index[0];
156 flat_index = flat_index * dims[i] + index[i];
158 return flat_index;
|
| /external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| D | reference_ops.h | 4351 int flat_index = i; in SelectTrueCoords() local
4353 int coord_j = flat_index / dims_to_count[j]; in SelectTrueCoords()
4355 flat_index %= dims_to_count[j]; in SelectTrueCoords()
|