HomeSort by relevance Sort by last modified time
    Searched refs:GetUniqueSlice (Results 1 - 10 of 10) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
buffer_assignment_test.cc 264 return *buffers.GetUniqueSlice(hlo, index).ConsumeValueOrDie().allocation();
    [all...]
buffer_assignment.cc 410 StatusOr<BufferAllocation::Slice> BufferAssignment::GetUniqueSlice(
444 return GetUniqueSlice(instruction, /*index=*/{});
450 return GetUniqueSlice(hlo_a, shape_index_a).ConsumeValueOrDie() ==
451 GetUniqueSlice(hlo_b, shape_index_b).ConsumeValueOrDie();
    [all...]
buffer_assignment.h 397 StatusOr<BufferAllocation::Slice> GetUniqueSlice(
399 // Like GetUniqueSlice but fixes the index to the top-level of the shape
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
ir_emitter_unnested.cc 417 auto output_data = assn.GetUniqueSlice(custom_call, {0}).ValueOrDie();
418 auto output_mean = assn.GetUniqueSlice(custom_call, {1}).ValueOrDie();
419 auto output_inv_stddev = assn.GetUniqueSlice(custom_call, {2}).ValueOrDie();
447 auto output_grad_data = assn.GetUniqueSlice(custom_call, {0}).ValueOrDie();
448 auto output_grad_scale = assn.GetUniqueSlice(custom_call, {1}).ValueOrDie();
450 assn.GetUniqueSlice(custom_call, {2}).ValueOrDie();
475 auto conv_result_slice = assn.GetUniqueSlice(custom_call, {0}).ValueOrDie();
476 auto scratch_slice = assn.GetUniqueSlice(custom_call, {1}).ValueOrDie();
500 auto a_buffer = assn.GetUniqueSlice(custom_call, {0}).ValueOrDie();
501 auto workspace_buffer = assn.GetUniqueSlice(custom_call, {1}).ValueOrDie()
    [all...]
ir_emitter.h 137 // A convenient helper for calling BufferAssignment::GetUniqueSlice.
141 .GetUniqueSlice(&hlo, index)
gpu_executable.cc 312 this->assignment_->GetUniqueSlice(src_hlo, sources[0]->index()));
hlo_to_ir_bindings.cc 104 buffer_assignment_->GetUniqueSlice(non_io_hlo, index);
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_executable.cc 236 this->assignment_->GetUniqueSlice(src, buffer_source->index()));
ir_emitter.h 190 // A convenient helper for calling BufferAssignment::GetUniqueSlice.
193 return assignment_.GetUniqueSlice(&hlo, index).ConsumeValueOrDie();
ir_emitter.cc 338 assignment_.GetUniqueSlice(infeed, {0}));
341 assignment_.GetUniqueSlice(infeed, {1}));
356 assignment_.GetUniqueSlice(infeed, {0, i}));
    [all...]

Completed in 298 milliseconds