Searched refs:ComputeBatchIndices (Results 1 – 2 of 2) sorted by relevance
57 ComputeBatchIndices(output_batch_size_, batch_bcast_->x_reshape(), in MatMulBCast()59 ComputeBatchIndices(output_batch_size_, batch_bcast_->y_reshape(), in MatMulBCast()
34 inline void ComputeBatchIndices(const int64 output_batch_size, in ComputeBatchIndices() function307 ComputeBatchIndices(output_batch_size_, reshape_[i], bcast_[i], in BCastList()