Searched refs:kInputIndex_Add (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_matmul_op_fused.cc | 143 const Tensor& add_tensor = MklGetInput(ctx, kInputIndex_Add); in Compute()
145 GetMklShape(ctx, kInputIndex_Add, &add_mkl_shape, native_format); in Compute()
149 kInputIndex_Add, kOutputIndex_Dst, in Compute()
153 ctx, kInputIndex_Add, kOutputIndex_Dst, in Compute()
297 const int kInputIndex_Add = 3; member in tensorflow::MklFusedMatMulOp
|
| D | mkl_conv_ops.cc | 877 const Tensor& add_tensor = MklGetInput(context, kInputIndex_Add); in AllocateOutputTensor()
879 GetMklShape(context, kInputIndex_Add, &add_mkl_shape, native_format); in AllocateOutputTensor()
883 kInputIndex_Add, kOutputIndex_Dst, in AllocateOutputTensor()
889 ForwardMklTensorInToOutWithMklShape(context, kInputIndex_Add, in AllocateOutputTensor()
962 const int kInputIndex_Add = 3; member in tensorflow::MklConvOp
|