Home
last modified time | relevance | path

Searched refs:input_dim_index (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/dtensor/mlir/expansions/
Dslice_spmd_expander.cc397 int input_dim_index = 0; in ApplyNewAndShrinkMasksToLayout() local
401 if (input_dim_index < input_rank && in ApplyNewAndShrinkMasksToLayout()
402 shrink_axis_mask[input_dim_index] == 1) { in ApplyNewAndShrinkMasksToLayout()
403 input_dim_index++; in ApplyNewAndShrinkMasksToLayout()
418 input_dim_index++; in ApplyNewAndShrinkMasksToLayout()
/external/tensorflow/tensorflow/compiler/xla/
Dshape_util.cc1288 for (int64_t input_dim_index : input_dim_indices) { in ReshapeLeavesDimensionsUnmodified() local
1292 unmodified_dims[i].first < input_dim_index) { in ReshapeLeavesDimensionsUnmodified()
1296 unmodified_dims[i].first != input_dim_index) { in ReshapeLeavesDimensionsUnmodified()