Searched refs:BufferAt (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | heap_simulator_test.cc | 333 const HloValue* BufferAt(const HloInstruction* instruction, in BufferAt() function in xla::__anonb650ca080111::HeapSimulatorTracker
340 const HloValue* buffer = BufferAt(instruction, index); in OffsetAt()
434 {kAlloc, tracker.BufferAt(param0, {})}, in TEST_F()
435 {kFree, tracker.BufferAt(param0, {})}, in TEST_F()
453 {kAlloc, tracker.BufferAt(paramA, {})}, in TEST_F()
454 {kAlloc, tracker.BufferAt(paramX, {})}, in TEST_F()
455 {kAlloc, tracker.BufferAt(mul, {})}, in TEST_F()
457 {kFree, tracker.BufferAt(paramA, {})}, in TEST_F()
458 {kFree, tracker.BufferAt(paramX, {})}, in TEST_F()
459 {kFree, tracker.BufferAt(mul, {})}, in TEST_F()
[all …]
|
| D | hlo_live_range_test.cc | 57 const HloValue* BufferAt(const HloInstruction* instruction, in BufferAt() function in xla::__anond24688290111::HloLiveRangeTest
65 auto* value = BufferAt(instruction, index);
|