Home
last modified time | relevance | path

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

/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/
H A 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()
H A 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()
H A Dbuffer_assignment.cc328 if (is_entry_computation_parameter()) { in ToString()
683 if (allocation.is_entry_computation_parameter()) { in ComputeSummaryStats()
H A Dhlo.proto561 bool is_entry_computation_parameter = 5; field
/aosp_15_r20/external/tensorflow/tensorflow/core/profiler/convert/
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/xla/transforms/
H A 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()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dbuffer_info_util.cc33 } else if (allocation.is_entry_computation_parameter()) { in CreateBufferInfosFromBufferAssignment()
H A Dcpu_executable.cc99 if (allocation.is_entry_computation_parameter()) { in MemoryForAllocation()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dgpu_executable.cc465 } else if (allocation.is_entry_computation_parameter()) { in BufferForAllocation()
514 if (allocation.is_entry_computation_parameter()) { in CheckAlignment()
H A Dir_emitter_unnested.cc521 if (alloc->is_entry_computation_parameter()) { in BuildKernelPrototype()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/pjrt/
H A Dtfrt_cpu_pjrt_client.cc1230 if (allocation.is_entry_computation_parameter()) { in MemoryForAllocation()