Searched refs:ForwardMklTensorInToOut (Results 1 – 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/mkl/ |
D | mkl_input_conversion_op.cc | 113 ForwardMklTensorInToOut(context, kInputIndex_0, kInputIndex_0); in Compute() 114 ForwardMklTensorInToOut(context, kInputIndex_1, kInputIndex_1); in Compute() 155 ForwardMklTensorInToOut(context, kInputIndex_1, kInputIndex_1); in Compute() 264 ForwardMklTensorInToOut(context, mkl_tensor_index, mkl_tensor_index); in Compute()
|
D | mkl_identity_op.cc | 43 ForwardMklTensorInToOut(context, kInputIdx, kOutputIdx); in Compute()
|
D | mkl_aggregate_ops.cc | 125 ForwardMklTensorInToOut(ctx, kSrc0Idx, kOutputIdx); in Compute()
|
/external/tensorflow/tensorflow/core/util/ |
D | mkl_util.h | 869 inline void ForwardMklTensorInToOut(OpKernelContext* context, int idx_in, in ForwardMklTensorInToOut() function
|