Searched refs:lt_input (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_conv_grad_bias_ops.cc | 161 dnnLayout_t lt_input, lt_outbackprop; member
173 lt_input = static_cast<dnnLayout_t>(input_shape.GetCurLayout()); in MklCreateInputLayouts()
175 CHECK_EQ(dnnLayoutCreate_F32(<_input, in_dims, in_sizes, in_strides), in MklCreateInputLayouts()
222 if (!dnnLayoutCompare_F32(lt_input, mkl_prim_internal_input)) { in MklPrepareConvolutionInputs()
223 CHECK_EQ(dnnConversionCreate_F32(&mkl_convert_input, lt_input, in MklPrepareConvolutionInputs()
246 if (!input_is_mkl) dnnLayoutDelete_F32(lt_input); in MklCleanup()
|
| D | mkl_conv_ops.cc | 701 dnnLayout_t lt_filter, lt_bias, lt_input; member
708 lt_input = static_cast<dnnLayout_t>(input_shape.GetCurLayout()); in MklCreateInputLayouts()
710 CHECK_EQ(dnnLayoutCreate_F32(<_input, in_dims, in_sizes, in_strides), in MklCreateInputLayouts()
755 !dnnLayoutCompare_F32(mkl_lt_internal_input, lt_input); in MklPrepareConvolutionInputs()
757 CHECK_EQ(dnnConversionCreate_F32(&mkl_prim_convert_input, lt_input, in MklPrepareConvolutionInputs()
828 if (!input_in_mkl_format) dnnLayoutDelete_F32(lt_input); in MklCleanup()
|