Searched defs:num_batch_dims (Results 1 – 12 of 12) sorted by relevance
34 const int num_batch_dims = a_shape.dimensions().size() - 2; in LuDecomposition() local
69 const int num_batch_dims = a_shape.dimensions().size() - 2; in SelfAdjointEig() local
119 const int64_t num_batch_dims = num_dims - 2; in HouseRow() local184 const int64_t num_batch_dims = num_dims - 2; in HouseCol() local258 const int64_t num_batch_dims = num_dims - 2; in HouseHolderBidiagonalization() local460 const int64_t num_batch_dims = num_dims - 2; in OneSidedJacobiUpdate() local839 const int64_t num_batch_dims = num_dims - 2; in SVD() local
55 int num_batch_dims = dim_nums->lhs_batch_dimensions_size(); in HandleCustomCall() local
176 size_t num_batch_dims = std::max(lhs_num_batch_dims, rhs_num_batch_dims); in For() local258 int64_t num_batch_dims = in For() local
213 const int64_t num_batch_dims = num_dims - 2; in QrBlock() local393 const int64_t num_batch_dims = num_dims - 2; in BuildQrDecomposition() local464 const int64_t num_batch_dims = num_dims - 2; in ProductOfElementaryHouseholderReflectors() local
44 const int64_t num_batch_dims = original_dnums.lhs_batch_dimensions_size(); in CanonicalizeDot() local
459 const int64_t num_batch_dims = num_dims - 2; in BuildEigh() local
2413 int64_t num_batch_dims = dim_numbers.rhs_batch_dimensions_size(); in EmitElementalDot() local
1328 int64_t num_batch_dims = in PotentiallyImplementedAsEigenMatmul() local1403 int64_t num_batch_dims = in EmitBatchDotOperation() local
342 num_batch_dims=1): argument754 num_batch_dims=1): argument
442 bool transpose_rhs, int64_t num_batch_dims, in BatchDot()6739 const int64_t num_batch_dims = num_dims - 2; in QRBlock() local