Searched refs:OffsetAt (Results 1 – 1 of 1) sorted by relevance
327 int64 OffsetAt(const HloInstruction* instruction, const ShapeIndex& index) { in OffsetAt() function in xla::__anon72fb25430111::HeapSimulatorTracker343 int64 offset_a = OffsetAt(instruction_a, index_a); in ExpectSharedBuffers()344 int64 offset_b = OffsetAt(instruction_b, index_b); in ExpectSharedBuffers()473 auto neg_buffer = tracker.OffsetAt(neg, {}); in TEST_F()474 int64 output_buffer_0 = tracker.OffsetAt(fusion, {0}); in TEST_F()475 int64 output_buffer_1 = tracker.OffsetAt(fusion, {1}); in TEST_F()