Searched refs:TFShapeToMklDnnDims (Results 1 – 10 of 10) sorted by relevance
131 memory::dims in_dims = TFShapeToMklDnnDims(in_tensor.shape()); in MKLTransposeND()132 memory::dims out_dims = TFShapeToMklDnnDims(out_tensor->shape()); in MKLTransposeND()
282 dst_dims = TFShapeToMklDnnDims(mkl_input_shapes[0].GetTfShape()); in Compute()287 dst_dims = TFShapeToMklDnnDims(input_tensors[0].shape()); in Compute()333 memory::dims src_dims = TFShapeToMklDnnDims(input_tensors[k].shape()); in Compute()507 auto src_dims = TFShapeToMklDnnDims(input_shapes[k].GetTfShape()); in FindMklCommonFormat()
63 auto src_dims = TFShapeToMklDnnDims(src_tf_shape); in Compute()
178 auto output_dims = TFShapeToMklDnnDims(shape_to); in Compute()
127 dims = TFShapeToMklDnnDims(src1_tensor.shape()); in Compute()
406 input_dims = TFShapeToMklDnnDims(input_tensor.shape()); in ComputeMklSlice()
477 src_dims = TFShapeToMklDnnDims(src_tensor.shape()); in Compute()590 src_dims = TFShapeToMklDnnDims(src_tensor.shape()); in Compute()
451 *output_dims_tf_order = TFShapeToMklDnnDims(out_shape);
1091 TFShapeToMklDnnDims(filter_tf_shape), in Compute()
1499 inline memory::dims TFShapeToMklDnnDims(const TensorShape& shape) { in TFShapeToMklDnnDims() function