Searched refs:source_slice (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | outfeed_thunk.cc | 86 BufferAllocation::Slice source_slice = source_slices_[index].slice; in ExecuteOnStream() local
87 if (!source_slice.allocation()) in ExecuteOnStream()
90 buffer_allocations.GetDeviceAddress(source_slice); in ExecuteOnStream()
|
| D | ir_emitter_unnested.cc | 3089 BufferAllocation::Slice source_slice, in EmitCollectivePermuteFromMlir()
3096 input.thunk_info, std::move(source_dest_pairs), source_slice, in EmitCollectivePermuteFromMlir()
3122 TF_ASSIGN_OR_RETURN(auto source_slice, GetAllocationSliceForMlir(operand)); in EmitNcclThunkFromMlir()
3126 /*source_buffer=*/source_slice, in EmitNcclThunkFromMlir()
|
| /external/tensorflow/tensorflow/python/kernel_tests/array_ops/ |
| D | gather_op_test.py | 167 source_slice = ((slice(None),) * outer_dims + (source_index,) +
169 correct_params_grad[dest_slice] += gather_grad[source_slice]
236 source_slice = ((slice(None),) * outer_dims + (source_index,) +
238 correct_params_grad[dest_slice] += gather_grad[source_slice]
|