Searched refs:dy_datas (Results 1 – 1 of 1) sorted by relevance
123 std::vector<Tensor> dy_datas(y_num); in ComputeTheoreticalJacobianTranspose() local125 dy_datas[i] = Tensor(ys[i].type(), y_shapes[i]); in ComputeTheoreticalJacobianTranspose()126 auto dy_data_flat = dy_datas[i].flat<Y_T>(); in ComputeTheoreticalJacobianTranspose()136 feed_list.insert({dys[i], dy_datas[i]}); in ComputeTheoreticalJacobianTranspose()146 auto dy_data_flat = dy_datas[y_idx].flat<Y_T>(); in ComputeTheoreticalJacobianTranspose()