Searched refs:SetTfLayout (Results 1 – 16 of 16) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util_test.cc | 36 a_mkldnn_shape.SetTfLayout(a_dims.size(), a_dims, memory::format::nchw); in TEST()
48 b_mkldnn_shape.SetTfLayout(b_dims.size(), b_dims, memory::format::nhwc); in TEST()
|
| D | mkl_util.h | 181 void SetTfLayout(const size_t dimension, const size_t* sizes, in SetTfLayout() function
634 inline void SetTfLayout(size_t dims, const memory::dims& sizes, in SetTfLayout() function
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_input_conversion_op.cc | 127 mkl_output_mkl_shape.SetTfLayout(input_shape_0.GetDimension(), in Compute()
225 mkl_output_mkl_shape.SetTfLayout(mkl_shape->GetDimension(), in Compute()
|
| D | mkl_aggregate_ops.cc | 223 output_mkl_shape.SetTfLayout(src1_dims_size, in Compute()
227 output_mkl_shape.SetTfLayout(src2_dims_size, in Compute()
|
| D | mkl_softmax_op.cc | 148 output_mkl_shape.SetTfLayout(output_dims.size(), output_dims, in Compute()
|
| D | mkl_reshape_op.cc | 190 mkl_shape_output.SetTfLayout(output_dims.size(), output_dims, in Compute()
|
| D | mkl_conv_grad_input_ops.cc | 428 diff_src_mkl_shape.SetTfLayout(bwd_diff_src_dims.size(), in Compute()
548 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
|
| D | mkl_avgpooling_op.cc | 128 mkl_out_shape.SetTfLayout(mkl_context.params.in_dim, in Compute()
319 mkl_out_shape.SetTfLayout(mkl_context.params.in_dim, in Compute()
|
| D | mkl_lrn_op.cc | 208 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
458 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
|
| D | mkl_relu_op.cc | 512 dnn_shape_dst.SetTfLayout(dnn_shape_src.GetDimension(), in Compute()
675 dnn_shape_diff_src.SetTfLayout(dnn_shape_src.GetDimension(), in Compute()
679 dnn_shape_diff_src.SetTfLayout( in Compute()
|
| D | mkl_pooling_ops_common.h | 652 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
704 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
|
| D | mkl_maxpooling_op.cc | 109 mkl_out_shape.SetTfLayout(mkl_context.params.in_dim, in Compute()
282 mkl_output_shape.SetTfLayout(mkl_context.params.in_dim, in Compute()
|
| D | mkl_conv_ops.cc | 629 mkl_output_mkl_shape.SetTfLayout(mkl_context.in_dims, mkl_context.out_sizes, in Compute()
652 mkl_filter_output_mkl_shape.SetTfLayout(filter.dims(), filter_sizes, in Compute()
1268 output_mkl_shape.SetTfLayout(output_dims_mkl_order.size(), in AllocateOutputTensor()
1344 filter_mkl_shape.SetTfLayout(filter_dims_tf_order.size(), in AllocateFilterOutputTensor()
|
| D | mkl_slice_op.cc | 462 output_mkl_shape->SetTfLayout(input_mkl_shape.GetDimension(), output_dims, in AllocateOutputTensor()
|
| D | mkl_fused_batch_norm_op.cc | 667 dnn_shape_dst.SetTfLayout(ndims, src_dims, format_m); in Compute()
998 dnn_shape_diff_src.SetTfLayout(src_dims.size(), src_dims, format_m); in Compute()
|
| D | mkl_concat_op.cc | 402 dnn_shape_dst.SetTfLayout(dst_dims.size(), dst_dims_in_nchw, in Compute()
|