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/cpu/
buffer_info_util.cc 31 } else if (allocation.is_entry_computation_parameter()) {
cpu_executable.cc 92 if (allocation.is_entry_computation_parameter()) {
239 if (!slice.allocation()->is_entry_computation_parameter()) {
  /external/tensorflow/tensorflow/compiler/xla/service/
buffer_assignment.h 101 return (is_entry_computation_parameter() &&
112 bool is_entry_computation_parameter() const { function in class:xla::BufferAllocation
216 return is_entry_computation_parameter() || maybe_live_out();
223 return !is_entry_computation_parameter() &&
hlo.proto 382 bool is_entry_computation_parameter = 5;
buffer_assignment_test.cc     [all...]
buffer_assignment.cc 290 if (is_entry_computation_parameter()) {
611 if (allocation.is_entry_computation_parameter()) {
    [all...]
  /external/tensorflow/tensorflow/compiler/xla/service/gpu/
buffer_allocations.cc 50 if (allocation.is_entry_computation_parameter()) {
gpu_executable.cc 254 if (allocation.is_entry_computation_parameter()) {
317 if (!slice.allocation()->is_entry_computation_parameter()) {
ir_emitter_unnested.cc 218 if (alloc->is_entry_computation_parameter()) {
    [all...]

Completed in 119 milliseconds