- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for getRhsContractingDimensions (0.25 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
<< filter_type << "\n"); return failure(); } const int rhs_rank = filter_type.cast<TensorType>().getRank(); const auto rhs_contracting_dim = dot_dimension_nums.getRhsContractingDimensions()[0]; if ((rhs_contracting_dim != rhs_rank - 1) && (rhs_contracting_dim != rhs_rank - 2)) { LLVM_DEBUG(llvm::dbgs() << "Not supported rhs contracting dim for dot_general.\n");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 22 09:00:19 UTC 2024 - 99.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
dot_dimension_numbers.getLhsContractingDimensions()); DotDimensionsInfo rhs_dot_dimensions_info( rhs_type, dot_dimension_numbers.getRhsBatchingDimensions(), dot_dimension_numbers.getRhsContractingDimensions()); // Transposes lhs shape to be in the order of {batch_dimensions, // out_dimensions, contracting dimensions}. llvm::SmallVector<int64_t, 4> lhs_permutation = Concat<int64_t>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
auto rhs_contracting_dimensions = builder_.CreateVector( mlir::GetVector<int64_t>(mlir::cast<mlir::vhlo::TensorV1Attr>( vhlo_op.getRhsContractingDimensions()), vhlo_type_converter)); BufferOffset<flatbuffers::Vector<unsigned int>> precision_config = 0; if (vhlo_op.getPrecisionConfig()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0)