Searched defs:result_buffers (Results 1 – 3 of 3) sorted by relevance
225 OpKernelContext* context, xla::ScopedShapedBuffer result_buffers, in UpdateOutputVariables()
405 std::vector<ScopedShapedBuffer> result_buffers; in ExecuteParallelAndRegisterResult() local
1436 auto result_buffers = CreateResultShapedBuffer(result_buffer_indices_, in ExecuteHelper() local