- Sort Score
- Result 10 results
- Languages All
Results 1 - 9 of 9 for FailureOr (0.42 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/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/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) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_cluster_formation.cc
} } } return WalkResult::advance(); }); return other_op; } // Collects ops that need to be moved behind the cluster due to data or control // dependencies. mlir::FailureOr<llvm::SmallSetVector<Operation*, 8>> CollectClusterSuccessorOps( Block* block, const OpSetVector& cluster_ops, const mlir::TF::SideEffectAnalysis::Info& side_effect_analysis, bool strict_clusters) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 22:03:30 UTC 2024 - 39.3K bytes - Viewed (0)