Searched defs:result_dim (Results 1 – 2 of 2) sorted by relevance
333 int64 result_dim; member370 for (int64 result_dim = result_shape.size() - 1; result_dim >= 0; in ComputeReshapePassthroughDimPairs() local779 auto map_passthrough_operand_dim_to_result_dim = [&](int64 result_dim) { in FoldReshapeOfGatherNoDegenerateDims()
35 const Index result_dim = input_dim - kernel_dim + 1; in IndexMapper() local