Home
last modified time | relevance | path

Searched refs:MakeSubBuffer (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/compiler/xrt/
Dxrt_util.cc143 XRTTupleAllocation::MakeSubBuffer(output.get(), {index - 1}, &tuple, in MakeOutput()
289 TF_RETURN_IF_ERROR(XRTTupleAllocation::MakeSubBuffer( in GetInputTupleAllocations()
358 TF_RETURN_IF_ERROR(XRTTupleAllocation::MakeSubBuffer( in CreateExecuteOutput()
Dxrt_state.h116 static Status MakeSubBuffer(XRTTupleAllocation* parent,
Dxrt_state.cc362 /*static*/ Status XRTTupleAllocation::MakeSubBuffer( in MakeSubBuffer() function in tensorflow::XRTTupleAllocation
/external/tensorflow/tensorflow/compiler/xrt/kernels/
Dxrt_state_ops.h404 ctx, XRTTupleAllocation::MakeSubBuffer(allocation.get(), shape_index, in Compute()
600 TF_RETURN_IF_ERROR(XRTTupleAllocation::MakeSubBuffer( in Compute()
Dtpu_execute_op.cc107 TF_RETURN_IF_ERROR(XRTTupleAllocation::MakeSubBuffer( in GetChainedOpInputs()
Dxrt_execute_op.cc115 TF_RETURN_IF_ERROR(XRTTupleAllocation::MakeSubBuffer( in GetChainedOpInputTuples()