Searched refs:SetTfDimOrder (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/core/util/ |
| D | mkl_util.h | 204 void SetTfDimOrder(const size_t dimension) { in SetTfDimOrder() function
214 void SetTfDimOrder(const size_t dimension, const size_t* tf_to_mkl_dim_map) { in SetTfDimOrder() function
224 void SetTfDimOrder(const size_t dimension, TensorFormat data_format) { in SetTfDimOrder() function
643 SetTfDimOrder(dims, format); in SetTfLayout()
670 inline void SetTfDimOrder(const size_t dimension, const mkldnn_dims_t map) { in SetTfDimOrder() function
677 inline void SetTfDimOrder(const size_t dimension, TensorFormat data_format) { in SetTfDimOrder() function
700 inline void SetTfDimOrder(const size_t dimension, memory::format format) { in SetTfDimOrder() function
702 SetTfDimOrder(dimension, data_format); in SetTfDimOrder()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | mkl_avgpooling_op.cc | 131 mkl_out_shape.SetTfDimOrder(mkl_context.params.in_dim, data_format_); in Compute()
322 mkl_out_shape.SetTfDimOrder(mkl_context.params.in_dim, data_format_); in Compute()
|
| D | mkl_maxpooling_op.cc | 112 mkl_out_shape.SetTfDimOrder(mkl_context.params.in_dim, data_format_); in Compute()
285 mkl_output_shape.SetTfDimOrder(mkl_context.params.in_dim, data_format_); in Compute()
|
| D | mkl_conv_ops.cc | 633 mkl_output_mkl_shape.SetTfDimOrder(mkl_context.in_dims, data_format_); in Compute()
655 mkl_filter_output_mkl_shape.SetTfDimOrder(mkl_context.filter_dims, in Compute()
|
| D | mkl_fused_batch_norm_op.cc | 999 dnn_shape_diff_src.SetTfDimOrder(src_dims.size(), tensor_format_); in Compute()
|