Home
last modified time | relevance | path

Searched refs:MklDnnDimsToTFShape (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/mkl/
Dmkl_dequantize_op.cc118 output_tf_shape = MklDnnDimsToTFShape(output_dims); in Compute()
Dmkl_slice_op.cc391 auto shape_to = MklDnnDimsToTFShape(size_dims); in ComputeMklSlice()
495 output_tf_shape = MklDnnDimsToTFShape(output_dims); in AllocateOutputTensor()
Dmkl_softmax_op.cc292 output_tf_shape = MklDnnDimsToTFShape(src_dims); in Compute()
Dmkl_pooling_ops_common.h552 output_tf_shape = MklDnnDimsToTFShape(output_dims_mkl_order); in AllocateEmptyOutputTensor()
568 output_tf_shape = MklDnnDimsToTFShape(output_dims_order); in AllocateEmptyOutputTensor()
Dmkl_concat_op.cc764 tf_shape_dst = MklDnnDimsToTFShape(dst_dims); in Compute()
801 tf_shape_dst = MklDnnDimsToTFShape(dst_dims); in Compute()
Dmkl_quantize_op.cc417 output_tf_shape = MklDnnDimsToTFShape(output_dims); in Compute()
Dmkl_conv_ops.cc548 TensorShape dst_tf_shape = MklDnnDimsToTFShape(dst_dims_tf_order); in Compute()
/external/tensorflow/tensorflow/core/util/
Dmkl_util.h1160 inline TensorShape MklDnnDimsToTFShape(const memory::dims& dims) { in MklDnnDimsToTFShape() function