Searched refs:y_batch_index (Results 1 – 3 of 3) sorted by relevance
100 const int64 y_batch_index = should_bcast ? y_batch_indices[i] : i; in Run() local105 auto y = Ty.template chip<0>(y_batch_index).conjugate(); in Run()108 auto y = Ty.template chip<0>(y_batch_index); in Run()144 const int64 y_batch_index = should_bcast ? y_batch_indices[i] : i;146 auto y = Ty.template chip<0>(y_batch_index);186 const int64 y_batch_index = should_bcast ? y_batch_indices[i] : i;188 auto y = ConstTensorSliceToEigenMatrix(in_y, y_batch_index);
73 const int64 y_batch_index = should_bcast ? y_batch_indices[i] : i; in Run() local75 auto rhs = ConstTensorSliceToEigenMatrix(in_y, y_batch_index); in Run()
92 const int64 y_batch_index = should_bcast ? y_batch_indices[i] : i; in Run() local94 auto rhs = ConstTensorSliceToEigenMatrix(in_y, y_batch_index); in Run()