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.h104 return (is_entry_computation_parameter() && in is_readonly()
115 bool is_entry_computation_parameter() const { in is_entry_computation_parameter() function
221 return is_entry_computation_parameter() || maybe_live_out(); in IsInputOrOutput()
228 return !is_entry_computation_parameter() && in IsPreallocatedTempBuffer()
Dbuffer_assignment_test.cc1406 EXPECT_FALSE(map_param_alloc.is_entry_computation_parameter()); in TEST_F()
1411 EXPECT_FALSE(map_root_alloc.is_entry_computation_parameter()); in TEST_F()
1417 EXPECT_TRUE(call_param_alloc.is_entry_computation_parameter()); in TEST_F()
1422 EXPECT_FALSE(call_root_alloc.is_entry_computation_parameter()); in TEST_F()
1428 EXPECT_TRUE(param_alloc.is_entry_computation_parameter()); in TEST_F()
1433 EXPECT_FALSE(map_alloc.is_entry_computation_parameter()); in TEST_F()
1463 EXPECT_TRUE(allocation.is_entry_computation_parameter()); in TEST_F()
2691 EXPECT_FALSE(root_alloc->is_entry_computation_parameter()); in TEST_F()
Dbuffer_assignment.cc343 if (is_entry_computation_parameter()) { in ToString()
703 if (allocation.is_entry_computation_parameter()) { in ComputeSummaryStats()
Dhlo.proto440 bool is_entry_computation_parameter = 5; field
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dbuffer_info_util.cc32 } else if (allocation.is_entry_computation_parameter()) { in CreateBufferInfosFromBufferAssignment()
Dcpu_executable.cc96 if (allocation.is_entry_computation_parameter()) { in MemoryForAllocation()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc1461 if (lhs->is_entry_computation_parameter() != in Initialize()
1462 rhs->is_entry_computation_parameter()) { in Initialize()
1463 return lhs->is_entry_computation_parameter() > in Initialize()
1464 rhs->is_entry_computation_parameter(); in Initialize()
1466 if (lhs->is_entry_computation_parameter()) { in Initialize()
1485 alloc->is_entry_computation_parameter()) { in Initialize()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.cc310 } else if (allocation.is_entry_computation_parameter()) { in BufferForAllocation()
356 if (allocation.is_entry_computation_parameter()) { in CheckAlignment()
Dir_emitter_unnested.cc587 if (alloc->is_entry_computation_parameter()) { in BuildKernelPrototype()