Home
last modified time | relevance | path

Searched refs:BufferSizeBytesFunction (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/compiler/xla/service/cpu/
Dcpu_compiler.cc602 ScheduleModule(module.get(), BufferSizeBytesFunction(), in RunHloPassesAndBufferAssignement()
611 BufferSizeBytesFunction(), memory_alignment, in RunHloPassesAndBufferAssignement()
718 ScheduleModule(module.get(), BufferSizeBytesFunction(), in RunBackend()
727 BufferSizeBytesFunction(), memory_alignment, in RunBackend()
915 ScheduleModule(module, BufferSizeBytesFunction())); in CompileAheadOfTime()
923 BufferSizeBytesFunction(), memory_alignment, in CompileAheadOfTime()
/external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/
Dcpu_noalias_test.cc68 backend().compiler()->BufferSizeBytesFunction(), in TEST_F()
/external/tensorflow/tensorflow/compiler/xla/service/
Dcompiler.h295 std::function<int64(const BufferValue&)> BufferSizeBytesFunction() { in BufferSizeBytesFunction() function
Dbuffer_assignment_test.cc94 backend().compiler()->BufferSizeBytesFunction(), in RunBufferAssignment()
104 backend().compiler()->BufferSizeBytesFunction(), in RunBufferAssignmentNoBuffersForConstants()
116 backend().compiler()->BufferSizeBytesFunction(), in RunBufferAssignmentNoBuffersReuseForAdd()
128 backend().compiler()->BufferSizeBytesFunction(), in RunColoredBufferAssignment()
141 backend().compiler()->BufferSizeBytesFunction(), in RunBufferAssignmentWithInstructionSequence()
152 backend().compiler()->BufferSizeBytesFunction(), in RunBufferAssignmentWithPresetAssignments()
2362 backend().compiler()->BufferSizeBytesFunction(), in TEST_F()