Searched refs:update_indices (Results 1 – 3 of 3) sorted by relevance
881 update_indices = update_col_indices887 update_indices = update_row_indices893 num_rows = math_ops.cast(array_ops.shape(update_indices)[0], dtypes.int64)926 row_wt, update_indices, partition_strategy="div")928 num_indices = array_ops.shape(update_indices)[0]958 update_indices,
554 std::vector<HloInstruction*> slice_indices, update_indices; in TEST_F() local559 update_indices.push_back( in TEST_F()575 slice, update_indices)); in TEST_F()
3843 std::vector<HloInstruction*> slice_indices, update_indices; in TEST_F() local3848 update_indices.push_back( in TEST_F()3864 slice, update_indices)); in TEST_F()