Searched refs:BufferSizeBytesFunction (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | cpu_compiler.cc | 639 ScheduleModule(module.get(), BufferSizeBytesFunction(), in RunBackend()
647 BufferSizeBytesFunction(), memory_alignment, in RunBackend()
824 ScheduleModule(module, BufferSizeBytesFunction())); in CompileAheadOfTime()
832 BufferSizeBytesFunction(), memory_alignment, in CompileAheadOfTime()
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/tests/ |
| D | cpu_noalias_test.cc | 66 backend().compiler()->BufferSizeBytesFunction(), in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | compiler.h | 250 std::function<int64(const BufferValue&)> BufferSizeBytesFunction() { in BufferSizeBytesFunction() function
|
| D | buffer_assignment_test.cc | 93 backend().compiler()->BufferSizeBytesFunction(), in RunBufferAssignment()
104 backend().compiler()->BufferSizeBytesFunction(), in RunBufferAssignmentNoBuffersForConstants()
120 backend().compiler()->BufferSizeBytesFunction(), in RunBufferAssignmentNoBuffersReuseForAdd()
133 backend().compiler()->BufferSizeBytesFunction(), in RunColoredBufferAssignment()
147 backend().compiler()->BufferSizeBytesFunction(), in RunBufferAssignmentWithInstructionSequence()
2214 backend().compiler()->BufferSizeBytesFunction(), in TEST_F()
|
| /external/tensorflow/tensorflow/compiler/xla/service/gpu/ |
| D | nvptx_compiler.cc | 670 BufferSizeBytesFunction(), in RunBackend()
|