Searched refs:SetTfLayout (Results 1 – 17 of 17) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util_test.cc | 33 a_mkldnn_shape.SetTfLayout(a_dims.size(), a_dims, in TEST()
46 b_mkldnn_shape.SetTfLayout(b_dims.size(), b_dims, in TEST()
|
| D | mkl_util.h | 436 inline void SetTfLayout(size_t dims, const memory::dims& sizes, in SetTfLayout() function
|
| /external/tensorflow/tensorflow/core/kernels/mkl/ |
| D | mkl_input_conversion_op.cc | 126 mkl_output_mkl_shape.SetTfLayout(input_shape_0.GetDimension(), in Compute()
227 mkl_output_mkl_shape.SetTfLayout(mkl_shape->GetDimension(), in Compute()
|
| D | mkl_dequantize_op_test.cc | 63 mkl_shape.SetTfLayout(4, {1, 2, 2, 2}, MklTensorFormat::FORMAT_NHWC); in CreateMklInput()
|
| D | mkl_dequantize_op.cc | 116 output_mkl_shape.SetTfLayout(src_mkl_shape.GetDimension(), in Compute()
|
| D | mkl_relu_op.cc | 505 dnn_shape_dst.SetTfLayout(dnn_shape_src.GetDimension(), in Compute()
509 dnn_shape_dst.SetTfLayout(src_tensor.dims(), in Compute()
717 dnn_shape_diff_src.SetTfLayout(dnn_shape_src.GetDimension(), in Compute()
721 dnn_shape_diff_src.SetTfLayout( in Compute()
|
| D | mkl_conv_grad_input_ops.cc | 449 diff_src_mkl_shape.SetTfLayout(bwd_diff_src_dims.size(), in Compute()
578 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
|
| D | mkl_aggregate_ops.cc | 226 output_mkl_shape.SetTfLayout(mkl_shape.GetDimension(), in Compute()
|
| D | mkl_softmax_op.cc | 288 output_mkl_shape.SetTfLayout(src_dims.size(), src_dims, layout_type); in Compute()
|
| D | mkl_lrn_op.cc | 204 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
466 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
|
| D | mkl_pooling_ops_common.h | 655 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
707 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
|
| D | mkl_quantize_op.cc | 411 output_mkl_shape.SetTfLayout(src_mkl_shape.GetDimension(), in Compute()
|
| D | mkl_slice_op.cc | 489 output_mkl_shape->SetTfLayout(input_mkl_shape.GetDimension(), output_dims, in AllocateOutputTensor()
|
| D | mkl_fused_batch_norm_op.cc | 861 dnn_shape_dst.SetTfLayout(ndims, src_dims, mkl_tensor_fmt); in Compute()
1259 dnn_shape_diff_src.SetTfLayout(src_dims.size(), src_dims, mkl_tensor_fmt); in Compute()
|
| D | mkl_matmul_ops_common.h | 416 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(),
|
| D | mkl_concat_op.cc | 729 dnn_shape_dst.SetTfLayout(dst_dims.size(), dst_dims_in_nchw, in Compute()
|
| D | mkl_conv_ops.cc | 1052 output_mkl_shape->SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
1215 filter_mkl_shape.SetTfLayout(filter_dims_tf_order.size(), in AllocateFilterOutputTensor()
|