- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for getLhsBatchingDimensions (0.47 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/optimize.cc
// rhs: [batch, contract dims, other dims] auto dim_nums = dot.getDotDimensionNumbers(); int batch_dims_count = dim_nums.getLhsBatchingDimensions().size(); int contracting_dims_count = dim_nums.getLhsContractingDimensions().size(); if (dim_nums.getLhsBatchingDimensions() != ArrayRef<int64_t>(range(0, batch_dims_count)) || dim_nums.getRhsBatchingDimensions() !=
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 26.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/attrs_and_constraints.cc
lhs_contracting_dims.size() == 1 && rhs_contracting_dims.size() == 1 && lhs_contracting_dims[0] == input_rank - 1; const bool is_not_batch_op = dot_dimension_numbers.getLhsBatchingDimensions().empty(); const bool has_proper_quantization_dimension = absl::c_find(rhs_contracting_dims, filter_rank) == rhs_contracting_dims.end();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo_conversions/dot_general.cc
ImplicitLocOpBuilder builder(loc, rewriter); // Collects lhs and rhs dimensions information. DotDimensionsInfo lhs_dot_dimensions_info( lhs_type, dot_dimension_numbers.getLhsBatchingDimensions(), dot_dimension_numbers.getLhsContractingDimensions()); DotDimensionsInfo rhs_dot_dimensions_info( rhs_type, dot_dimension_numbers.getRhsBatchingDimensions(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 19.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
"only one use of requantization.\n"); return failure(); } const int num_lhs_batching_dims = dot_dimension_nums.getLhsBatchingDimensions().size(); const int num_lhs_contracting_dims = dot_dimension_nums.getLhsContractingDimensions().size(); if (num_lhs_batching_dims > 3) {
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
ImplicitLocOpBuilder builder(loc, rewriter); // Collects lhs and rhs dimensions information. DotDimensionsInfo lhs_dot_dimensions_info( lhs_type, dot_dimension_numbers.getLhsBatchingDimensions(), dot_dimension_numbers.getLhsContractingDimensions()); DotDimensionsInfo rhs_dot_dimensions_info( rhs_type, dot_dimension_numbers.getRhsBatchingDimensions(),
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 lhs_batching_dimensions = builder_.CreateVector( mlir::GetVector<int64_t>(mlir::cast<mlir::vhlo::TensorV1Attr>( vhlo_op.getLhsBatchingDimensions()), vhlo_type_converter)); auto rhs_batching_dimensions = builder_.CreateVector( mlir::GetVector<int64_t>(mlir::cast<mlir::vhlo::TensorV1Attr>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0)