Searched refs:dnn_shape_dst (Results 1 – 3 of 3) sorted by relevance
450 MklDnnShape dnn_shape_dst; in Compute() local455 dnn_shape_dst.SetMklTensor(false); in Compute()458 dnn_shape_dst); in Compute()499 dnn_shape_dst.SetMklTensor(true); in Compute()501 dnn_shape_dst.SetMklLayout(&dst_pd); in Compute()502 dnn_shape_dst.SetElemType(MklDnnType<T>()); in Compute()504 dnn_shape_dst.SetTfLayout(dnn_shape_src.GetDimension(), in Compute()508 dnn_shape_dst.SetTfLayout(src_tensor.dims(), in Compute()516 dnn_shape_dst.SetMklTensor(false); in Compute()523 dnn_shape_dst); in Compute()[all …]
721 MklDnnShape dnn_shape_dst; in Compute() local724 dnn_shape_dst.SetMklTensor(true); in Compute()725 dnn_shape_dst.SetMklLayout(&dst_pd); in Compute()726 dnn_shape_dst.SetElemType(MklDnnType<T>()); in Compute()727 dnn_shape_dst.SetTfLayout(dst_dims.size(), dst_dims_in_nchw, in Compute()731 dnn_shape_dst); in Compute()737 if (dnn_shape_dst.IsMklTensor()) in Compute()738 dst_md = dnn_shape_dst.GetMklLayout(); in Compute()760 MklDnnShape dnn_shape_dst; in Compute() local763 dnn_shape_dst.SetMklTensor(false); in Compute()[all …]
846 MklDnnShape dnn_shape_dst; in Compute() local848 dnn_shape_dst.SetMklTensor(true); in Compute()850 dnn_shape_dst.SetMklLayout(&dst_pd); in Compute()851 dnn_shape_dst.SetElemType(MklDnnType<T>()); in Compute()854 dnn_shape_dst.SetTfLayout(ndims, src_dims, mkl_tensor_fmt); in Compute()857 tf_shape_dst = dnn_shape_dst.GetTfShape(); in Compute()860 dnn_shape_dst, native_format); in Compute()945 MklDnnShape dnn_shape_dst; in HandleEmptyInput() local946 dnn_shape_dst.SetMklTensor(false); in HandleEmptyInput()948 dnn_shape_dst, native_format); in HandleEmptyInput()