Searched refs:kInputIndex_Filter (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dmkl_conv_grad_input_ops.cc370 const int kInputIndex_Filter = 1, kInputIndex_InputSizes = 0, member in class:tensorflow::MklConv2DCustomBackpropInputOp
397 return GetTfShape(context, kInputIndex_Filter);
H A Dmkl_conv_ops.cc518 const Tensor& filter_tensor = MklGetInput(context, kInputIndex_Filter);
522 GetMklShape(context, kInputIndex_Filter, &filter_mkl_shape);
537 auto filter_tf_shape = GetTfShape(context, kInputIndex_Filter);
661 const int kInputIndex_Src = 0, kInputIndex_Filter = 1, kInputIndex_Bias = 2; member in class:tensorflow::MklConv2DOp

Completed in 185 milliseconds