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

  /external/tensorflow/tensorflow/compiler/xla/service/
hlo.proto 205 bool is_entry_computation_parameter = 5;
buffer_assignment.h 84 bool is_entry_computation_parameter() const { function in class:xla::BufferAllocation
180 return is_entry_computation_parameter() || maybe_live_out();
187 return !is_entry_computation_parameter() &&
buffer_assignment_test.cc     [all...]
buffer_assignment.cc 124 if (is_entry_computation_parameter()) {
414 if (allocation.is_entry_computation_parameter()) {
670 if (allocation->is_entry_computation_parameter()) {
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
gpu_executable.cc 264 if (allocation.is_entry_computation_parameter()) {
314 CHECK(!slice.allocation()->is_entry_computation_parameter());
  /external/tensorflow/tensorflow/compiler/xla/service/cpu/
cpu_executable.cc 88 if (allocation.is_entry_computation_parameter()) {
275 CHECK(!slice.allocation()->is_entry_computation_parameter());
parallel_cpu_executable.cc 338 if (allocation.is_entry_computation_parameter()) {
492 CHECK(!slice.allocation()->is_entry_computation_parameter());
cpu_compiler.cc     [all...]
  /external/tensorflow/tensorflow/compiler/tf2xla/
xla_jit_compiled_cpu_function.cc 62 if (allocation.is_entry_computation_parameter() ||

Completed in 226 milliseconds