/external/tensorflow/tensorflow/core/kernels/ |
mkl_conv_grad_input_ops.cc | 250 dnnLayout_t lt_filter, lt_outbackprop; member in struct:tensorflow::MklConv2DCustomBackpropInputOp::__anon39840 257 lt_filter = (dnnLayout_t)filter_shape.GetCurLayout(); 259 CHECK_EQ(dnnLayoutCreate_F32(<_filter, in_dims, filter_size, 293 if (!dnnLayoutCompare_F32(mkl_lt_filter_internal, lt_filter)) { 295 CHECK_EQ(dnnConversionCreate_F32(&mkl_convert_filter, lt_filter, 348 if (!filter_in_mkl_format) dnnLayoutDelete_F32(lt_filter);
|
mkl_conv_ops.cc | 348 dnnLayout_t lt_filter, lt_bias, lt_input; member in struct:tensorflow::MklConv2DOp::__anon39841 361 CHECK_EQ(dnnLayoutCreate_F32(<_filter, filter_dims, filter_sizes, 426 !dnnLayoutCompare_F32(mkl_lt_internal_filter, lt_filter); 428 CHECK_EQ(dnnConversionCreate_F32(&mkl_prim_convert_filter, lt_filter, 476 dnnLayoutDelete_F32(lt_filter); [all...] |