- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 21 for lhs_contracting_dimensions (0.33 sec)
-
tensorflow/compiler/mlir/lite/tests/flatbuffer2mlir/vhlo.mlir
%0 = "vhlo.dot_general_v1"(%arg0, %arg1) <{ lhs_batching_dimensions = #vhlo.tensor_v1<dense<0> : tensor<1xi64>>, lhs_contracting_dimensions = #vhlo.tensor_v1<dense<2> : tensor<1xi64>>, rhs_batching_dimensions = #vhlo.tensor_v1<dense<0> : tensor<1xi64>>, rhs_contracting_dimensions = #vhlo.tensor_v1<dense<0> : tensor<1xi64>>,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 14 19:15:40 UTC 2024 - 31.9K bytes - Viewed (1) -
tensorflow/compiler/mlir/lite/stablehlo/tests/legalize_hlo.mlir
%0 = "mhlo.dot_general"(%arg0, %arg1) { dot_dimension_numbers = #mhlo.dot< lhs_batching_dimensions = [0], lhs_contracting_dimensions = [1, 2], rhs_batching_dimensions = [0], rhs_contracting_dimensions = [1, 3] >, precision_config = [#mhlo<precision DEFAULT>, #mhlo<precision DEFAULT>] } : (tensor<3x2x6x5x1xf32>, tensor<3x2x4x6xf32>) -> tensor<3x5x1x4xf32>
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 29 07:26:59 UTC 2024 - 340.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/schema/schema.fbs
DEFAULT, HIGH, HIGHEST, } table StablehloDotGeneralOptions{ lhs_batching_dimensions : [long]; rhs_batching_dimensions : [long]; lhs_contracting_dimensions : [long]; rhs_contracting_dimensions : [long]; precision_config : [StablehloPrecisionConfig]; } table StablehloReduceWindowOptions{ window_dimensions : [long]; window_strides : [long];
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 03 18:01:23 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/schema/schema_generated.h
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 1M bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
vhlo_type_converter)); auto lhs_contracting_dimensions = builder_.CreateVector( mlir::GetVector<int64_t>(mlir::cast<mlir::vhlo::TensorV1Attr>( vhlo_op.getLhsContractingDimensions()), vhlo_type_converter)); auto rhs_contracting_dimensions = builder_.CreateVector(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (2) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
// (C1) size(lhs_batching_dimensions) = size(rhs_batching_dimensions). if (num_lhs_contracting_dims != 1) { // Check one side is enough since // (C2) size(lhs_contracting_dimensions) = // size(rhs_contracting_dimensions). LLVM_DEBUG(llvm::dbgs() << "Failed to match contract dimension for " "quantized dot_general.\n"); return failure(); }
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/tf2xla/transforms/legalize_tf.cc
/*lhs_batching_dimensions=*/batch_dimensions, /*rhs_batching_dimensions=*/batch_dimensions, /*lhs_contracting_dimensions=*/lhs_contracting_dimensions, /*rhs_contracting_dimensions=*/rhs_contracting_dimensions); // TODO(silvasean): Emit shape checks for contracting dimensions. // (The batch dimensions are checked by the broadcasting logic) rewriter.replaceOpWithNewOp<DotGeneralOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 291.8K bytes - Viewed (1) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
/*rhs_batching_dimensions=*/{}, /*lhs_contracting_dimensions=*/ {lhs_rank == 1 ? 0 : 1}, /*rhs_contracting_dimensions=*/{0}); return ConvertDot( rewriter, dot_op.getLhs(), dot_op.getRhs(), dot_dimension_numbers,
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/tf2xla/tests/legalize-tf.mlir
// CHECK-SAME: dot_dimension_numbers = #mhlo.dot< // CHECK-NOT: lhs_batching_dimensions = // CHECK-NOT: rhs_batching_dimensions = // CHECK-SAME: lhs_contracting_dimensions = [1] // CHECK-SAME: rhs_contracting_dimensions = [0] // CHECK-SAME: precision_config = []
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 06 18:46:23 UTC 2024 - 335.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/optimize.cc
op, op.getType(), op.getLhs(), op.getRhs(), mhlo::DotDimensionNumbersAttr::get( op.getContext(), /*lhsBatchingDimensions=*/{}, /*rhsBatchingDimensions=*/{}, /*lhsContractingDimensions=*/{lhs_type.getRank() - 1}, /*rhsContractingDimensions=*/{0}), op.getPrecisionConfigAttr()); return success(); } // Convert reshape(dot_general(reshape(%y), %z)) to
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 26.9K bytes - Viewed (0)