Searched refs:dnn_shape_diff_src (Results 1 – 2 of 2) sorted by relevance
609 MklDnnShape dnn_shape_diff_src; in Compute() local612 dnn_shape_diff_src.SetMklTensor(false); in Compute()615 tf_shape_diff_src, dnn_shape_diff_src); in Compute()711 dnn_shape_diff_src.SetMklTensor(true); in Compute()712 dnn_shape_diff_src.SetMklLayout(&diff_src_pd); in Compute()713 dnn_shape_diff_src.SetElemType(MklDnnType<T>()); in Compute()715 dnn_shape_diff_src.SetTfLayout(dnn_shape_src.GetDimension(), in Compute()719 dnn_shape_diff_src.SetTfLayout( in Compute()726 dnn_shape_diff_src.SetMklTensor(false); in Compute()734 AllocateOutputSetMklShape(context, diff_src_index, dnn_shape_diff_src); in Compute()[all …]
1244 MklDnnShape dnn_shape_diff_src; in Compute() local1246 dnn_shape_diff_src.SetMklTensor(true); in Compute()1248 dnn_shape_diff_src.SetMklLayout(&diff_src_pd); in Compute()1249 dnn_shape_diff_src.SetElemType(MklDnnType<T>()); in Compute()1250 dnn_shape_diff_src.SetTfLayout(src_dims.size(), src_dims, mkl_tensor_fmt); in Compute()1251 dnn_shape_diff_src.SetTfDimOrder(src_dims.size(), tensor_format_); in Compute()1254 tf_shape_diff_src = dnn_shape_diff_src.GetTfShape(); in Compute()1257 tf_shape_diff_src, dnn_shape_diff_src, in Compute()1321 MklDnnShape dnn_shape_diff_src; in HandleEmptyInput() local1322 dnn_shape_diff_src.SetMklTensor(false); in HandleEmptyInput()[all …]