Searched refs:conv_utl (Results 1 – 3 of 3) sorted by relevance
/external/tensorflow/tensorflow/core/kernels/ |
D | mkl_conv_grad_input_ops.cc | 366 MklDnnConvUtil conv_utl(context, this->strides_, this->padding_, in Compute() local 368 conv_utl.GetConvFwdSizesInMklOrder( in Compute() 393 conv_utl.GetInputSizeInMklOrder(diff_dst_tf_shape, &diff_dst_dims); in Compute()
|
D | mkl_conv_grad_filter_ops.cc | 430 MklDnnConvUtil conv_utl(context, this->strides_, this->padding_, in Compute() local 432 conv_utl.GetConvFwdSizesInMklOrder( in Compute() 447 conv_utl.GetInputSizeInMklOrder(diff_dst_tf_shape, &diff_dst_dims); in Compute()
|
D | mkl_conv_ops.cc | 960 MklDnnConvUtil conv_utl(context, strides_, padding_, data_format_, in Compute() local 964 conv_utl.GetConvFwdSizesInMklOrder( in Compute() 1070 conv_utl.GetBiasSizeInMklOrder(kInputIndex_Bias, &bias_dims); in Compute()
|