Home
last modified time | relevance | path

Searched refs:SetTfLayout (Results 1 – 17 of 17) sorted by relevance

/external/tensorflow/tensorflow/core/util/
Dmkl_util_test.cc35 a_mkldnn_shape.SetTfLayout(a_dims.size(), a_dims, MKL_TENSOR_FORMAT_NCHW); in TEST()
47 b_mkldnn_shape.SetTfLayout(b_dims.size(), b_dims, MKL_TENSOR_FORMAT_NHWC); in TEST()
Dmkl_util.h441 inline void SetTfLayout(size_t dims, const memory::dims& sizes, in SetTfLayout() function
/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_input_conversion_op.cc126 mkl_output_mkl_shape.SetTfLayout(input_shape_0.GetDimension(), in Compute()
227 mkl_output_mkl_shape.SetTfLayout(mkl_shape->GetDimension(), in Compute()
Dmkl_dequantize_op_test.cc72 mkl_shape.SetTfLayout(4, {1, 2, 2, 2}, MKL_TENSOR_FORMAT_NHWC); in CreateMklInput()
Dmkl_dequantize_op.cc112 output_mkl_shape.SetTfLayout(src_mkl_shape.GetDimension(), in Compute()
Dmkl_conv_grad_input_ops.cc447 diff_src_mkl_shape.SetTfLayout(bwd_diff_src_dims.size(), in Compute()
574 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
Dmkl_relu_op.cc504 dnn_shape_dst.SetTfLayout(dnn_shape_src.GetDimension(), in Compute()
508 dnn_shape_dst.SetTfLayout(src_tensor.dims(), in Compute()
715 dnn_shape_diff_src.SetTfLayout(dnn_shape_src.GetDimension(), in Compute()
719 dnn_shape_diff_src.SetTfLayout( in Compute()
Dmkl_aggregate_ops.cc225 output_mkl_shape.SetTfLayout(mkl_shape.GetDimension(), in Compute()
Dmkl_softmax_op.cc288 output_mkl_shape.SetTfLayout(src_dims.size(), src_dims, layout_type); in Compute()
Dmkl_lrn_op.cc203 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
465 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
Dmkl_pooling_ops_common.h655 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
707 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
Dmkl_quantize_op.cc411 output_mkl_shape.SetTfLayout(src_mkl_shape.GetDimension(), in Compute()
Dmkl_slice_op.cc488 output_mkl_shape->SetTfLayout(input_mkl_shape.GetDimension(), output_dims, in AllocateOutputTensor()
Dmkl_fused_batch_norm_op.cc854 dnn_shape_dst.SetTfLayout(ndims, src_dims, mkl_tensor_fmt); in Compute()
1250 dnn_shape_diff_src.SetTfLayout(src_dims.size(), src_dims, mkl_tensor_fmt); in Compute()
Dmkl_matmul_ops_common.h401 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
Dmkl_concat_op.cc727 dnn_shape_dst.SetTfLayout(dst_dims.size(), dst_dims_in_nchw, in Compute()
Dmkl_conv_ops.cc866 output_mkl_shape->SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
1025 filter_mkl_shape.SetTfLayout(filter_dims_tf_order.size(), in AllocateFilterOutputTensor()