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