Searched refs:MutableResult (Results 1 – 5 of 5) sorted by relevance
98 for (auto& pair : result.MutableResult()->buffers()) { in AllocateOutputMemoryWithInputReuse()
448 for (auto& p : result.MutableResult()->buffers().leaves()) { in ExecuteAsyncOnStreamImpl()457 for (auto& p : result.MutableResult()->buffers()) { in ExecuteAsyncOnStreamImpl()
192 ScopedShapedBuffer* MutableResult() { return &result_; } in MutableResult() function
241 for (auto& p : result.MutableResult()->buffers()) { in CreateResultShapedBuffer()
271 xla::ScopedShapedBuffer* shaped_buffer = run_result.MutableResult(); in CreateOutputTuple()