Searched refs:arg_buffer (Results 1 – 3 of 3) sorted by relevance
355 auto arg_buffer = LiteralToShapedBuffer(arg_literal); in XLA_TEST_F() local357 ScopedShapedBuffer result = ExecuteLocallyOrDie(computation, {&arg_buffer}); in XLA_TEST_F()385 auto arg_buffer = LiteralToShapedBuffer(arg_literal); in XLA_TEST_F() local387 ScopedShapedBuffer result_0 = ExecuteLocallyOrDie(computation, {&arg_buffer}); in XLA_TEST_F()437 auto arg_buffer = LiteralToShapedBuffer(arg_literal); in XLA_TEST_F() local439 ScopedShapedBuffer result = ExecuteLocallyOrDie(computation, {&arg_buffer}); in XLA_TEST_F()491 auto arg_buffer = LiteralToShapedBuffer(arg_literal); in XLA_TEST_F() local493 ScopedShapedBuffer result = ExecuteLocallyOrDie(computation, {&arg_buffer}); in XLA_TEST_F()535 auto arg_buffer = LiteralToShapedBuffer(arg_literal); in XLA_TEST_F() local537 ScopedShapedBuffer result = ExecuteLocallyOrDie(computation, {&arg_buffer}); in XLA_TEST_F()
439 arg_buffer = self.backend.buffer_from_pyval(arg)440 arg_buffer.delete()442 compiled_c.execute([arg_buffer])453 arg_buffer = self.backend.buffer_from_pyval(arg)454 arg_buffer.block_host_until_ready()
355 %define arg_buffer 8 + stack_offset547 mov buffer, [frame + arg_buffer]