OpenGrok
Home
Sort by relevance
Sort by last modified time
Full Search
Definition
Symbol
File Path
History
|
|
Help
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