HomeSort by relevance Sort by last modified time
    Searched defs:buffer_assignment (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_executable.h 25 #include "tensorflow/compiler/xla/service/buffer_assignment.h"
90 const BufferAssignment& buffer_assignment() const { return *assignment_; } function in class:xla::cpu::CpuExecutable
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
ir_emitter_context.h 20 #include "tensorflow/compiler/xla/service/buffer_assignment.h"
34 const BufferAssignment* buffer_assignment,
38 buffer_assignment_(buffer_assignment),
47 const BufferAssignment& buffer_assignment() const { function in class:xla::gpu::IrEmitterContext
  /external/tensorflow/tensorflow/compiler/tf2xla/
xla_jit_compiled_cpu_function.cc 54 const xla::BufferAssignment& buffer_assignment) {
56 buffer_assignment.Allocations();
74 const xla::BufferAssignment& buffer_assignment) {
76 buffer_assignment.GetUniqueTopLevelOutputSlice());
152 const xla::BufferAssignment& buffer_assignment = local
153 cpu_executable->buffer_assignment();
159 ComputeTempSizes(buffer_assignment));
161 ComputeResultIndex(buffer_assignment));

Completed in 83 milliseconds