Searched refs:ForwardTfTensorInToOut (Results 1 – 4 of 4) sorted by relevance
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
H A D | mkl_input_conversion_op.cc | 75 ForwardTfTensorInToOut(context, kInputIndex_0, kInputIndex_0); in Compute() 76 ForwardTfTensorInToOut(context, kInputIndex_1, kInputIndex_1); in Compute() 264 ForwardTfTensorInToOut(context, tf_tensor_index, tf_tensor_index); in Compute()
|
H A D | mkl_identity_op.cc | 39 ForwardTfTensorInToOut(context, kInputIdx, kOutputIdx); in Compute()
|
H A D | mkl_aggregate_ops.cc | 126 ForwardTfTensorInToOut(ctx, kSrc0Idx, kOutputIdx); in Compute()
|
/aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
H A D | mkl_util.h | 893 inline void ForwardTfTensorInToOut(OpKernelContext* context, int idx_in, in ForwardTfTensorInToOut() function
|