Searched refs:scoped_buffers (Results 1 – 7 of 7) sorted by relevance
395 OpKernelContext* context, xla::ScopedShapedBuffer scoped_buffers, in AllocateOutputTensors() argument400 VLOG(4) << "Output buffers: " << scoped_buffers.ToString(); in AllocateOutputTensors()405 xla::ShapeUtil::TupleElementCount(scoped_buffers.on_host_shape()); in AllocateOutputTensors()422 xla::ShapeUtil::GetSubshape(scoped_buffers.on_host_shape(), {i}); in AllocateOutputTensors()433 TF_RET_CHECK(scoped_buffers.on_host_shape().IsTuple()); in AllocateOutputTensors()434 TF_RET_CHECK(!xla::ShapeUtil::IsNestedTuple(scoped_buffers.on_host_shape())); in AllocateOutputTensors()440 absl::make_unique<OutputBuffers>(std::move(scoped_buffers), allocator); in AllocateOutputTensors()
1349 ScopedDeadlockReportBuffers scoped_buffers; in DeadlockCheck() local1350 DeadlockReportBuffers *b = scoped_buffers.b; in DeadlockCheck()
1352 ScopedDeadlockReportBuffers scoped_buffers; in DeadlockCheck() local1353 DeadlockReportBuffers *b = scoped_buffers.b; in DeadlockCheck()
1361 ScopedDeadlockReportBuffers scoped_buffers; in DeadlockCheck() local1362 DeadlockReportBuffers *b = scoped_buffers.b; in DeadlockCheck()