Searched refs:gather_index (Results 1 – 2 of 2) sorted by relevance
365 for (auto gather_index : dense_spec.final_shape_gather_indices) { in ValidateStridedSliceOp() local366 if (gather_index >= 0) { in ValidateStridedSliceOp()367 final_shape->AddDim(processing_shape->dim_size(gather_index)); in ValidateStridedSliceOp()368 } else if (gather_index == kNewAxis) { in ValidateStridedSliceOp()
73 gather_index = array_ops.boolean_mask(iota_r1, selected_index)74 gathered_input = array_ops.gather(padding_input, gather_index, axis=1)