Home
last modified time | relevance | path

Searched refs:is_entry_computation_parameter (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.h101 return (is_entry_computation_parameter() && in is_readonly()
112 bool is_entry_computation_parameter() const { in is_entry_computation_parameter() function
216 return is_entry_computation_parameter() || maybe_live_out(); in IsInputOrOutput()
223 return !is_entry_computation_parameter() && in IsPreallocatedTempBuffer()
Dbuffer_assignment_test.cc1234 EXPECT_FALSE(map_param_alloc.is_entry_computation_parameter()); in TEST_F()
1239 EXPECT_FALSE(map_root_alloc.is_entry_computation_parameter()); in TEST_F()
1245 EXPECT_TRUE(call_param_alloc.is_entry_computation_parameter()); in TEST_F()
1250 EXPECT_FALSE(call_root_alloc.is_entry_computation_parameter()); in TEST_F()
1256 EXPECT_TRUE(param_alloc.is_entry_computation_parameter()); in TEST_F()
1261 EXPECT_FALSE(map_alloc.is_entry_computation_parameter()); in TEST_F()
1291 EXPECT_TRUE(allocation.is_entry_computation_parameter()); in TEST_F()
2510 EXPECT_FALSE(root_alloc->is_entry_computation_parameter()); in TEST_F()
Dbuffer_assignment.cc290 if (is_entry_computation_parameter()) { in ToString()
611 if (allocation.is_entry_computation_parameter()) { in ComputeSummaryStats()
Dhlo.proto382 bool is_entry_computation_parameter = 5; field
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dbuffer_info_util.cc31 } else if (allocation.is_entry_computation_parameter()) { in CreateBufferInfosFromBufferAssignment()
Dcpu_executable.cc92 if (allocation.is_entry_computation_parameter()) { in CreateBufferTable()
239 if (!slice.allocation()->is_entry_computation_parameter()) { in CreateResultShapedBuffer()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.cc254 if (allocation.is_entry_computation_parameter()) { in ExecuteOnStream()
317 if (!slice.allocation()->is_entry_computation_parameter()) { in ExecuteOnStream()
Dbuffer_allocations.cc50 if (allocation.is_entry_computation_parameter()) { in Build()
Dir_emitter_unnested.cc218 if (alloc->is_entry_computation_parameter()) { in BuildKernelPrototype()