Searched defs:num_batch_dims (Results 1 – 10 of 10) sorted by relevance
34 const int num_batch_dims = a_shape.dimensions().size() - 2; in LuDecomposition() local
118 const int64 num_batch_dims = num_dims - 2; in HouseRow() local183 const int64 num_batch_dims = num_dims - 2; in HouseCol() local257 const int64 num_batch_dims = num_dims - 2; in HouseHolderBidiagonalization() local459 const int64 num_batch_dims = num_dims - 2; in OneSidedJacobiUpdate() local824 const int64 num_batch_dims = num_dims - 2; in SVD() local
429 const int64 num_batch_dims = num_dims - 2; in SelfAdjointEig() local
42 const int64 num_batch_dims = original_dnums.lhs_batch_dimensions_size(); in CanonicalizeDot() local
189 const int64 num_batch_dims = num_dims - 2; in QrBlock() local369 const int64 num_batch_dims = num_dims - 2; in BuildQrDecomposition() local
47 int64 num_batch_dims = in CanFoldOperandsIntoDot() local
2220 int64 num_batch_dims = dim_numbers.rhs_batch_dimensions_size(); in EmitElementalDot() local
1134 int64 num_batch_dims = in EmitBatchDotOperation() local
194 num_batch_dims=1): argument601 num_batch_dims=1): argument
540 bool transpose_rhs, int64_t num_batch_dims, in BatchDot()5825 const int64_t num_batch_dims = num_dims - 2; in QRBlock() local