Searched refs:tf_shape_diff_src (Results 1 – 2 of 2) sorted by relevance
608 TensorShape tf_shape_diff_src; in Compute() local613 tf_shape_diff_src = MklGetInput(context, diff_src_index).shape(); in Compute()615 tf_shape_diff_src, dnn_shape_diff_src); in Compute()724 tf_shape_diff_src.AddDim(diff_src_pd.get_size() / sizeof(T)); in Compute()727 tf_shape_diff_src = src_tensor.shape(); in Compute()733 tf_shape_diff_src, &diff_src_tensor)); in Compute()
1245 TensorShape tf_shape_diff_src; in Compute() local1252 tf_shape_diff_src.AddDim(diff_src_pd.get_size() / sizeof(T)); 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()