Searched refs:TFShapeToMklDnnDims (Results 1 – 13 of 13) sorted by relevance
183 auto lhs_dims = TFShapeToMklDnnDims(lhs_shape); in CreateMatMulParams()184 auto rhs_dims = TFShapeToMklDnnDims(rhs_shape); in CreateMatMulParams()185 auto out_dims = TFShapeToMklDnnDims(out_shape); in CreateMatMulParams()
132 memory::dims in_dims = TFShapeToMklDnnDims(in_tensor.shape()); in MKLTransposeND()133 memory::dims out_dims = TFShapeToMklDnnDims(out_tensor->shape()); in MKLTransposeND()
70 : TFShapeToMklDnnDims(src_tensor.shape()); in Compute()
583 dst_dims = TFShapeToMklDnnDims(mkl_input_shapes[0].GetTfShape()); in Compute()588 dst_dims = TFShapeToMklDnnDims(input_tensors[0].shape()); in Compute()638 memory::dims src_dims = TFShapeToMklDnnDims(s_shape); in Compute()880 auto src_dims = TFShapeToMklDnnDims(input_shapes[k].GetTfShape()); in FindMklCommonFormat()
188 src_dims = TFShapeToMklDnnDims(src_tf_shape); in Compute()189 weight_dims = TFShapeToMklDnnDims(weight_tf_shape); in Compute()
206 auto dims = TFShapeToMklDnnDims(src_tensor.shape()); in Compute()
221 src_dims = TFShapeToMklDnnDims(src_tf_shape); in Compute()
470 src_dims = TFShapeToMklDnnDims(src_tensor.shape()); in Compute()509 TFShapeToMklDnnDims(src_tensor.shape()), in Compute()626 src_dims = TFShapeToMklDnnDims(src_tensor.shape()); in Compute()
338 : TFShapeToMklDnnDims(src_tensor.shape()); in Compute()
424 input_dims = TFShapeToMklDnnDims(input_tensor.shape()); in ComputeMklSlice()
460 *output_dims_tf_order = TFShapeToMklDnnDims(out_shape);
672 TFShapeToMklDnnDims(filter_tf_shape), in Compute()
1070 inline memory::dims TFShapeToMklDnnDims(const TensorShape& shape) { in TFShapeToMklDnnDims() function