Searched refs:CreateMemoryMinimizingSequence (Results 1 - 7 of 7) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/
H A Dhlo_scheduling.h49 CreateMemoryMinimizingSequence(
54 StatusOr<std::vector<const HloInstruction*>> CreateMemoryMinimizingSequence(
H A Dhlo_scheduling.cc407 StatusOr<std::vector<const HloInstruction*>> CreateMemoryMinimizingSequence( function in namespace:xla::__anon25650
460 CreateMemoryMinimizingSequence(const HloModule& module, function in namespace:xla
469 CreateMemoryMinimizingSequence(*computation, *points_to_analysis,
475 StatusOr<std::vector<const HloInstruction*>> CreateMemoryMinimizingSequence( function in namespace:xla
482 return CreateMemoryMinimizingSequence(computation, *points_to_analysis,
H A Dhlo_ordering_test.cc69 CreateMemoryMinimizingSequence(*module, [](const LogicalBuffer& buffer) {
H A Dhlo_rematerialization.cc1217 TF_ASSIGN_OR_RETURN(*sequence, CreateMemoryMinimizingSequence(
H A Dbuffer_assignment_test.cc1511 CreateMemoryMinimizingSequence(*module, ByteSizeOf).ConsumeValueOrDie();
1866 CreateMemoryMinimizingSequence(*module, ByteSizeOf).ConsumeValueOrDie();
/external/tensorflow/tensorflow/compiler/xla/service/gpu/
H A Dhlo_schedule.cc201 CreateMemoryMinimizingSequence(
/external/tensorflow/tensorflow/compiler/xla/service/cpu/
H A Dcpu_compiler.cc638 CreateMemoryMinimizingSequence(*module, BufferSizeBytesFunction()));
817 CreateMemoryMinimizingSequence(*module, BufferSizeBytesFunction()));

Completed in 3119 milliseconds