Searched defs:reduction_index (Results 1 – 5 of 5) sorted by relevance
169 for (int64_t reduction_index = 0; reduction_index < reduction_iter_size; in Compute() local
1144 def wrapper_fn(index, reduction_index): argument1230 def intermediate_fn(index, reduction_index): argument1570 def intermediate_loader_fn(index, reduction_index, loader): argument
5099 def loader(index, reduction_index): argument
578 const int32_t reduction_index = reduction_indices[i]; in ProcessTensorFlowReductionOperator() local
1902 const T reduction_index = reduction_indices(i); in ReductionShapeHelper() local