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

  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
hlo_to_ir_bindings.cc 77 ->GetUniqueTopLevelSlice(non_io_hlo->LatestNonGteAncestor())
ir_emitter_unnested.cc 695 buffer_assignment.GetUniqueTopLevelSlice(copy->operand(0)).ok()) {
746 .GetUniqueTopLevelSlice(tuple_element)
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/
buffer_assignment_test.cc 246 *buffers.GetUniqueTopLevelSlice(hlo).ConsumeValueOrDie().allocation();
268 return *buffers.GetUniqueTopLevelSlice(hlo)
318 assignment.GetUniqueTopLevelSlice(instruction).ConsumeValueOrDie());
324 if (a_slices.contains(assignment.GetUniqueTopLevelSlice(instruction)
    [all...]
buffer_assignment.h 401 StatusOr<BufferAllocation::Slice> GetUniqueTopLevelSlice(
403 // Like GetUniqueTopLevelSlice but returns the slice for the output of the
buffer_assignment.cc 442 StatusOr<BufferAllocation::Slice> BufferAssignment::GetUniqueTopLevelSlice(
490 return GetUniqueTopLevelSlice(
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
ir_emitter.cc 125 assignment_.GetUniqueTopLevelSlice(computation->root_instruction()));
130 assignment_.GetUniqueTopLevelSlice(param));
    [all...]

Completed in 193 milliseconds