Searched refs:MklDnnDimsToTFShape (Results 1 – 6 of 6) sorted by relevance
153 output_tf_shape = MklDnnDimsToTFShape(output_dims); in Compute()
383 auto shape_to = MklDnnDimsToTFShape(size_dims); in ComputeMklSlice()469 output_tf_shape = MklDnnDimsToTFShape(output_dims); in AllocateOutputTensor()
562 output_tf_shape = MklDnnDimsToTFShape(output_dims_mkl_order); in AllocateEmptyOutputTensor()569 output_tf_shape = MklDnnDimsToTFShape(output_dims_NHWC_order); in AllocateEmptyOutputTensor()
407 tf_shape_dst = MklDnnDimsToTFShape(dst_dims); in Compute()
972 TensorShape dst_tf_shape = MklDnnDimsToTFShape(dst_dims_tf_order); in Compute()
1571 inline TensorShape MklDnnDimsToTFShape(const memory::dims& dims) { in MklDnnDimsToTFShape() function