Searched defs:num_batch_dims (Results 1 – 6 of 6) sorted by relevance
123 const int64 num_batch_dims = num_dims - 2; in HouseRow() local189 const int64 num_batch_dims = num_dims - 2; in HouseCol() local263 const int64 num_batch_dims = num_dims - 2; in HouseHolderBidiagonalization() local468 const int64 num_batch_dims = num_dims - 2; in OneSidedJacobiUpdate() local837 const int64 num_batch_dims = num_dims - 2; in SVD() local
167 const int64 num_batch_dims = num_dims - 2; in QRBlock() local344 const int64 num_batch_dims = num_dims - 2; in QRDecomposition() local
429 const int64 num_batch_dims = num_dims - 2; in SelfAdjointEig() local
46 const int64 num_batch_dims = dnums.lhs_batch_dimensions_size(); in DecomposeBatchDot() local168 const int64 num_batch_dims = original_dnums.lhs_batch_dimensions_size(); in CanonicalizeDot() local
2161 int64 num_batch_dims = dim_numbers.rhs_batch_dimensions_size(); in EmitElementalDot() local
938 int64 num_batch_dims = in EmitBatchDotOperation() local