Home
last modified time | relevance | path

Searched refs:mkl_context (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dmkl_conv_grad_bias_ops.cc65 MklConvBackBiasOpContext mkl_context; in Compute() local
67 GetMklShape(context, 0, &mkl_context.input_shape); in Compute()
68 bool input_is_mkl = mkl_context.input_shape.IsMklTensor(); in Compute()
72 context, mkl_context.input_shape.GetDimension() == 4, in Compute()
81 mkl_context.c_size = mkl_context.input_shape.GetSizes()[MklDims::C]; in Compute()
83 mkl_context.c_size = GetTensorDim(input, data_format_, 'C'); in Compute()
89 TensorShape output_shape{mkl_context.c_size}; in Compute()
97 mkl_context.in_dims = 4; in Compute()
100 mkl_context.in_sizes[MklDims::W] = in Compute()
101 mkl_context.input_shape.GetSizes()[MklDims::W]; in Compute()
[all …]
Dmkl_avgpooling_op.cc69 MklAvgPoolingOpContext mkl_context; in Compute() local
71 GetMklShape(context, 0, &mkl_context.input_shape); in Compute()
72 bool input_in_mkl_format = mkl_context.input_shape.IsMklTensor(); in Compute()
75 mkl_context.params.in_dim = tensor_in.dims(); in Compute()
77 mkl_context.params.in_dim = mkl_context.input_shape.GetDimension(); in Compute()
85 &mkl_context.input_shape); in Compute()
89 ExtractMklOpParams(context, data_format_, pool_params, &mkl_context.params); in Compute()
92 mkl_context.MklCreateLayoutsAndPrimitives(context, in Compute()
98 AllocTmpBuffer(context, &workspace_tensor, mkl_context.lt_workspace, in Compute()
101 if (mkl_context.convert_input != nullptr) { in Compute()
[all …]
Dmkl_maxpooling_op.cc77 MklMaxPoolingOpContext mkl_context; in Compute() local
80 GetMklShape(context, 0, &mkl_context.input_shape); in Compute()
81 bool input_in_mkl_format = mkl_context.input_shape.IsMklTensor(); in Compute()
83 mkl_context.params.in_dim = 4; in Compute()
94 &mkl_context.input_shape); in Compute()
99 ExtractMklOpParams(context, data_format_, pool_params, &mkl_context.params); in Compute()
101 mkl_context.MklCreateLayoutsAndPrimitives(context); in Compute()
108 mkl_out_shape.SetMklLayout(mkl_context.prim_pooling_fwd, dnnResourceDst); in Compute()
109 mkl_out_shape.SetTfLayout(mkl_context.params.in_dim, in Compute()
110 mkl_context.params.out_sizes, in Compute()
[all …]
Dmkl_conv_ops.cc441 MklConv2DOpContext mkl_context; in Compute() local
443 GetMklShape(context, 0, &(mkl_context.input_shape)); in Compute()
444 bool input_in_mkl_format = mkl_context.input_shape.IsMklTensor(); in Compute()
477 input_in_mkl_format ? GetMklTensorDim(mkl_context.input_shape, 'C') in Compute()
489 input_in_mkl_format ? GetMklTensorDim(mkl_context.input_shape, 'H') in Compute()
501 input_in_mkl_format ? GetMklTensorDim(mkl_context.input_shape, 'W') in Compute()
512 input_in_mkl_format ? GetMklTensorDim(mkl_context.input_shape, 'N') in Compute()
559 mkl_context.in_dims = input_in_mkl_format in Compute()
560 ? mkl_context.input_shape.GetDimension() in Compute()
562 mkl_context.filter_dims = filter.dims(); in Compute()
[all …]