Searched refs:x_batch_index (Results 1 – 2 of 2) sorted by relevance
100 const int64 x_batch_index = should_bcast ? x_batch_indices[i] : i; in Run() local103 auto x = Tx.template chip<0>(x_batch_index); in Run()137 const int64 x_batch_index = should_bcast ? x_batch_indices[i] : i;139 auto x = Tx.template chip<0>(x_batch_index);178 const int64 x_batch_index = should_bcast ? x_batch_indices[i] : i;180 auto x = ConstTensorSliceToEigenMatrix(in_x, x_batch_index);561 const int64 x_batch_index = should_bcast ? x_batch_indices[i] : i;564 auto x = Tx.template chip<0>(x_batch_index);
91 const int64 x_batch_index = should_bcast ? x_batch_indices[i] : i; in Run() local93 auto matrix = ConstTensorSliceToEigenMatrix(in_x, x_batch_index); in Run()