Searched refs:x_batch_index (Results 1 – 3 of 3) sorted by relevance
99 const int64 x_batch_index = should_bcast ? x_batch_indices[i] : i; in Run() local102 auto x = Tx.template chip<0>(x_batch_index); in Run()143 const int64 x_batch_index = should_bcast ? x_batch_indices[i] : i;145 auto x = Tx.template chip<0>(x_batch_index);185 const int64 x_batch_index = should_bcast ? x_batch_indices[i] : i;187 auto x = ConstTensorSliceToEigenMatrix(in_x, x_batch_index);
72 const int64 x_batch_index = should_bcast ? x_batch_indices[i] : i; in Run() local74 auto matrix = ConstTensorSliceToEigenMatrix(in_x, x_batch_index); in Run()
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()