- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 11 for FailureOr (0.44 sec)
-
tensorflow/compiler/mlir/quantization/stablehlo/passes/quantization_patterns.h
: OpRewritePattern<RootOpT>(context, /*benefit=*/300) {} private: // Collects all candidate ops for quantization, which are the // `dequantize_op`'s users. FailureOr<SmallVector<Operation*>> CollectCandidateOps( DequantizeOpT dequantize_op) const { auto users = dequantize_op->getResult(0).getUsers(); return SmallVector<Operation*>(users.begin(), users.end()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/attrs_and_constraints.h
<< T::getOperationName() << ").\n"); return failure(); } } FailureOr<int32_t> CastI64ToI32(int64_t value); // Tries to cast an array of int64 to int32. If any of the element in the // array is not in the range of int32, returns failure(). FailureOr<SmallVector<int32_t>> CastI64ArrayToI32( ArrayRef<int64_t> int64_array);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/fold_broadcast_pass.cc
if (splatLhs && splatRhs) { auto signedLhs = addSign(splatLhs.getSplatValue<ValType>(), etype); auto signedRhs = addSign(splatRhs.getSplatValue<ValType>(), etype); FailureOr<decltype(signedLhs)> result(Convert()(signedLhs, signedRhs)); return succeeded(result) ? SplatElementsAttr::get(type, *result) : Attribute(); } SmallVector<ValType, 6> values;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_ops_to_mhlo.cc
using quant::tensorflow::IsTFQintType; #define GEN_PASS_DEF_CONVERTTFQUANTOPSTOMHLO #include "tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/passes.h.inc" template <typename UniformQuantizedOp> FailureOr<TensorType> GetUniformQuantizedType( UniformQuantizedOp op, Type original_type, TypedValue<TensorType> scales_value, TypedValue<TensorType> zero_points_value, FloatType expressed_type,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 30.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/ops/stablehlo_op_quant_spec_test.cc
} )mlir"; const OwningOpRef<ModuleOp> module_op = ParseModuleOpString(kXlaCallModuleOpWithPerTensorQuantizedConvolution); ASSERT_TRUE(module_op); const FailureOr<TF::XlaCallModuleOp> xla_call_module_op = FindFirstOpFromMainFunc<TF::XlaCallModuleOp>(*module_op); ASSERT_TRUE(succeeded(xla_call_module_op)); const std::unique_ptr<OpQuantSpec> op_quant_spec =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 04 07:19:09 UTC 2024 - 14.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/merge_initializer_function_ops_to_main.cc
// Returns initializer_type -> init_func_op mapping from the session_init_op's // initializers. The initializer functions are validated for whether it can be // moved to the main function. Returns failure() iff validation fails. FailureOr<absl::flat_hash_map<std::string, func::FuncOp>> GetInitFuncOps( ModuleOp module_op) { absl::flat_hash_map<std::string, func::FuncOp> init_func_ops;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun May 12 12:54:52 UTC 2024 - 15.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_deserialization.cc
// naming conflicts. But we need to rename the function calls inside the // stablehlo functions as well. So we first do this renaming in the stablehlo // module itself without inserting into the tf module. FailureOr<StringAttr> RenameStablehloFunctions( MLIRContext *context, SymbolTableCollection &symbol_tables, ModuleOp tf_module, ModuleOp stablehlo_module) { SymbolTable &tf_symbol_table = symbol_tables.getSymbolTable(tf_module);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/lift_as_function_call_test.cc
return %0 : tensor<1x1x4xf32> } )mlir"; const OwningOpRef<ModuleOp> module_op = ParseModuleOpString(kXlaCallModuleOpWithQuantizationMethodAttr); ASSERT_TRUE(module_op); FailureOr<TF::XlaCallModuleOp> xla_call_module_op = FindFirstOpFromMainFunc<TF::XlaCallModuleOp>(*module_op); ASSERT_TRUE(succeeded(xla_call_module_op));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 10 04:07:09 UTC 2024 - 26.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/saved_model_export.cc
/*add_passes_func=*/ [dup_constants = export_opts.duplicate_shape_determining_constants]( PassManager& pm) { AddExportPasses(pm, dup_constants); }, ctx, module_op)); FailureOr<SmallVector<AssetFileDef>> asset_file_defs = quant::ConvertAssetArgs(module_op); if (failed(asset_file_defs)) { return absl::InternalError("Failed to convert asset args."); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Mar 20 11:11:25 UTC 2024 - 12.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/attrs_and_constraints_test.cc
TEST_F(AttrsAndConstraintsTest, I64ArrayInI32RangeAreCastedCorrectly) { const SmallVector<int64_t> array_i64 = {llvm::minIntN(32), -2, -1, 0, 1, 2, llvm::maxIntN(32)}; FailureOr<SmallVector<int32_t>> array_i32 = CastI64ArrayToI32(array_i64); EXPECT_TRUE(succeeded(array_i32)); EXPECT_THAT( *array_i32, ElementsAreArray({static_cast<int32_t>(llvm::minIntN(32)), -2, -1, 0, 1,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 14 17:10:32 UTC 2024 - 22.9K bytes - Viewed (0)