Searched refs:kOutputIndex_Dst (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_op_fused.cc | 149 kInputIndex_Add, kOutputIndex_Dst, in Compute()
153 ctx, kInputIndex_Add, kOutputIndex_Dst, in Compute()
159 AllocateOutputSetMklShape(ctx, kOutputIndex_Dst, &dst_tensor, in Compute()
298 const int kOutputIndex_Dst = 0; member in tensorflow::MklFusedMatMulOp
|
| D | mkl_conv_ops.cc | 560 AllocateOutputSetMklShape(context, kOutputIndex_Dst, &dst_tensor, in Compute()
883 kInputIndex_Add, kOutputIndex_Dst, in AllocateOutputTensor()
890 kOutputIndex_Dst, output_tensor, in AllocateOutputTensor()
894 AllocateOutputSetMklShape(context, kOutputIndex_Dst, output_tensor, in AllocateOutputTensor()
927 AllocateOutputSetMklShape(context, kOutputIndex_Dst, output_tensor, in AllocateOutputTensor()
963 const int kOutputIndex_Dst = 0, kOutputIndex_Filter = 1; member in tensorflow::MklConvOp
|