Searched refs:x_datas (Results 1 – 1 of 1) sorted by relevance
| /aosp_15_r20/external/tensorflow/tensorflow/cc/framework/ |
| H A D | gradient_checker.cc | 107 const std::vector<Tensor>& x_datas, const OutputList& ys, in ComputeTheoreticalJacobianTranspose() argument
134 feed_list.insert({xs[i], x_datas[i]}); in ComputeTheoreticalJacobianTranspose()
187 const OutputList& ys, std::vector<Tensor>* x_datas, in EvaluateGraph() argument
191 for (int i = 0; i < x_datas->size(); i++) { in EvaluateGraph()
192 feed_list.insert({xs[i], (*x_datas)[i]}); in EvaluateGraph()
197 for (int x_idx = 0; x_idx < x_datas->size(); x_idx++) { in EvaluateGraph()
199 if (y_data.SharesBufferWith((*x_datas)[x_idx])) { in EvaluateGraph()
217 std::vector<Tensor>* x_datas, in ComputeNumericJacobianTranspose() argument
229 auto x_data_flat = (*x_datas)[x_idx].flat<X_T>(); in ComputeNumericJacobianTranspose()
245 TF_RETURN_IF_ERROR(EvaluateGraph(&session, xs, ys, x_datas, &y_pos)); in ComputeNumericJacobianTranspose()
[all …]
|