Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/compiler/xla/service/
Dbuffer_assignment.h102 return (is_entry_computation_parameter() && in is_readonly()
113 bool is_entry_computation_parameter() const { in is_entry_computation_parameter() function
219 return is_entry_computation_parameter() || maybe_live_out(); in IsInputOrOutput()
226 return !is_entry_computation_parameter() && in IsPreallocatedTempBuffer()
Dbuffer_assignment_test.cc1425 EXPECT_FALSE(map_param_alloc.is_entry_computation_parameter()); in TEST_F()
1430 EXPECT_FALSE(map_root_alloc.is_entry_computation_parameter()); in TEST_F()
1436 EXPECT_TRUE(call_param_alloc.is_entry_computation_parameter()); in TEST_F()
1441 EXPECT_FALSE(call_root_alloc.is_entry_computation_parameter()); in TEST_F()
1447 EXPECT_TRUE(param_alloc.is_entry_computation_parameter()); in TEST_F()
1452 EXPECT_FALSE(map_alloc.is_entry_computation_parameter()); in TEST_F()
1484 EXPECT_FALSE(map_param_alloc.is_entry_computation_parameter()); in TEST_F()
1489 EXPECT_FALSE(map_root_alloc.is_entry_computation_parameter()); in TEST_F()
1519 EXPECT_TRUE(allocation.is_entry_computation_parameter()); in TEST_F()
2828 EXPECT_FALSE(root_alloc->is_entry_computation_parameter()); in TEST_F()
Dbuffer_assignment.cc328 if (is_entry_computation_parameter()) { in ToString()
683 if (allocation.is_entry_computation_parameter()) { in ComputeSummaryStats()
Dhlo.proto561 bool is_entry_computation_parameter = 5; field
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dbuffer_info_util.cc33 } else if (allocation.is_entry_computation_parameter()) { in CreateBufferInfosFromBufferAssignment()
Dcpu_executable.cc99 if (allocation.is_entry_computation_parameter()) { in MemoryForAllocation()
/external/tensorflow/tensorflow/core/profiler/convert/
Dhlo_proto_to_memory_visualization_utils.cc64 if (buffer_allocation->is_entry_computation_parameter()) { in GetAllocationGroupName()
188 if (proto.is_entry_computation_parameter()) { in Convert()
231 if (buffer_allocation->is_entry_computation_parameter()) { in NoteSpecialAllocations()
624 !buffer_allocation->is_entry_computation_parameter() && in GetHeapSimulatorTraceIdFromBufferAllocationIndex()
/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
Dmhlo_to_lhlo_with_xla.cc1583 if (lhs->is_entry_computation_parameter() != in Initialize()
1584 rhs->is_entry_computation_parameter()) { in Initialize()
1585 return lhs->is_entry_computation_parameter() > in Initialize()
1586 rhs->is_entry_computation_parameter(); in Initialize()
1588 if (lhs->is_entry_computation_parameter()) { in Initialize()
1644 if (alloc->is_entry_computation_parameter()) { in Initialize()
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
Dgpu_executable.cc465 } else if (allocation.is_entry_computation_parameter()) { in BufferForAllocation()
514 if (allocation.is_entry_computation_parameter()) { in CheckAlignment()
Dir_emitter_unnested.cc521 if (alloc->is_entry_computation_parameter()) { in BuildKernelPrototype()
/external/tensorflow/tensorflow/compiler/xla/pjrt/
Dtfrt_cpu_pjrt_client.cc1230 if (allocation.is_entry_computation_parameter()) { in MemoryForAllocation()