- Sort Score
- Result 10 results
- Languages All
Results 1 - 8 of 8 for mhlo (1.61 sec)
-
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
// Converts a mhlo.reduce op with a mlho binary operation into a tensorflow // reduction operation. If the initial value can be ignored, then convert it // into a single TfReduceOp. Otherwise, convert it into a TfReduceOp followed by // a TfBinaryOp. // For example: // 1) A mhlo::ReduceOp on value `x` with a mhlo::AndOp and a constant initial // value `true` is converted to a TF::Any on value `x`.
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/tensorflow/transforms/tf_passes.td
and sets the shape of these inputs to be dynamic shaped. This will ensure that the generated HLO program is correctly reflecting the dynamic shape. }]; // Required for mhlo bounded shape extension. let dependentDialects = ["mhlo::MhloDialect"]; let constructor = "TFTPU::CreateTPUAnnotateDynamicShapeInputsPass()"; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:18:05 UTC 2024 - 99.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/tpu_rewrite.mlir
return %3: tensor<512xi32> } func.func private @_func(%arg0: tensor<?xi32, #mhlo.type_extensions<bounds = [512]>> {mhlo.sharding = "\08\01\1A\01\01\22\01\00"}) -> (tensor<512xi32>) { %0 = "tf.A"(%arg0) {} : (tensor<?xi32, #mhlo.type_extensions<bounds = [512]>>) -> tensor<512xi32> return %0 : tensor<512xi32> } } // -----
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 172.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/uniform_quantized_stablehlo_to_tfl_pass.cc
// Splits hybrid quantized `stablehlo.dot_general` into `tfl.dequantize` and // float `stablehlo.dot_general` op. Legalization of float // `stablehlo.dot_general` op relies on existing passes for conversion of // StableHLO -> MHLO -> TF -> TFL. class RewriteHybridQuantizedDotGeneralOp : public OpRewritePattern<stablehlo::DotGeneralOp> { public: using OpRewritePattern<stablehlo::DotGeneralOp>::OpRewritePattern;
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/tensorflow/transforms/shape_inference.cc
for (int i = 0; i < input_ty.getShape().size(); i++) { if (hlo::isDynamicDimSize(input_ty.getShape()[i])) { new_bounds[i] = new_shape[i]; new_shape[i] = ShapedType::kDynamic; } } new_arg_type = tensorflow::GetTypeFromTFTensorShape( new_shape, element_type, mhlo::TypeExtensionsAttr::get(context, new_bounds)); } } return new_arg_type;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
else if (llvm::isa<mlir::vhlo::BooleanV1Attr>(attr)) flex_builder->Bool(name.c_str(), attr.cast<mlir::vhlo::BooleanV1Attr>().getValue()); else if (llvm::isa<mlir::vhlo::StringV1Attr>(attr)) flex_builder->String( name.c_str(), attr.cast<mlir::vhlo::StringV1Attr>().getValue().str()); else if (llvm::isa<mlir::vhlo::IntegerV1Attr>(attr)) flex_builder->Int(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/tests/shape_inference.mlir
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jan 23 17:24:10 UTC 2024 - 167.4K bytes - Viewed (0) -
src/cmd/vendor/rsc.io/markdown/entity.go
"м": "\u043c", "—": "\u2014", "∡": "\u2221", "𝔪": "\U0001d52a", "℧": "\u2127", "µ": "\u00b5", "∣": "\u2223", "*": "\u002a",
Registered: Wed Jun 12 16:32:35 UTC 2024 - Last Modified: Wed Jan 24 13:01:26 UTC 2024 - 101K bytes - Viewed (0)