Home
last modified time | relevance | path

Searched refs:x_batch_index (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Dbatch_matmul_op_impl.h100 const int64 x_batch_index = should_bcast ? x_batch_indices[i] : i; in Run() local
103 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);
Dmatrix_triangular_solve_op_impl.h91 const int64 x_batch_index = should_bcast ? x_batch_indices[i] : i; in Run() local
93 auto matrix = ConstTensorSliceToEigenMatrix(in_x, x_batch_index); in Run()