- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 27 for FailureOr (0.16 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/cc/convert_asset_args_test.cc
return } } )mlir"); FailureOr<SmallVector<AssetFileDef>> asset_file_defs = ConvertAssetArgs(*module_op); EXPECT_TRUE(succeeded(asset_file_defs)); EXPECT_THAT(*asset_file_defs, SizeIs(1));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/attrs_and_constraints.cc
return failure(); } return static_cast<int32_t>(value); } FailureOr<SmallVector<int32_t>> CastI64ArrayToI32( const ArrayRef<int64_t> int64_array) { SmallVector<int32_t> int32_array{}; int32_array.reserve(int64_array.size()); for (const int64_t i64 : int64_array) { FailureOr<int32_t> cast_i32 = CastI64ToI32(i64); if (failed(cast_i32)) return failure();
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/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/tensorflow/transforms/xla_call_module_serialization.cc
constexpr llvm::StringRef kCalledIndexAttrName = "called_index"; constexpr llvm::StringRef kCalledFuncAttrName = "called_func"; // Converts `called_func` attributes in custom call ops back to `called_index`. FailureOr<ArrayAttr> DesymbolizeCustomCallCalledIndex(ModuleOp module) { Builder builder(module.getContext()); SmallVector<Attribute> function_list; llvm::DenseMap<SymbolRefAttr, int> called_indexes;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.h
// Returns a failure() on error, otherwise returns true to indicate that it // reached convergence, false otherwise. // If input shapes are provided, first refines the `main` function using // InferShapeForFunction. FailureOr<bool> InferModuleShape(ModuleOp module, int64_t max_iterations = 10, ArrayRef<TypeID> ops_to_skip = {}, ArrayRef<ArrayRef<int64_t>> input_shapes = {});
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 12:49:45 UTC 2024 - 3.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/utils/tf_type_utils.h
#include "mlir/Support/LogicalResult.h" // from @llvm-project namespace mlir::quant::tensorflow { // GetDenseAttrFromTensorProtoAttr returns DenseElementsAttr from tensor proto. FailureOr<mlir::DenseElementsAttr> GetDenseAttrFromTensorProtoAttr( llvm::StringRef mangled_tensor_proto, TensorType result_tensor_type); // Check if a type is TF qint type. bool IsTFQintType(Type type);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 08:32:43 UTC 2024 - 1.7K 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/quantization/tensorflow/cc/convert_asset_args.h
// "tf_saved_model.index_path", if the bound input is attached to the // `tf_saved_model::AssetOp`. // * Strips the "assets/" prefix of the filename when setting it to // `AssetFileDef`. FailureOr<SmallVector<tensorflow::AssetFileDef>> ConvertAssetArgs( ModuleOp module_op); } // namespace mlir::quant
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Dec 19 01:45:41 UTC 2023 - 1.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/set_tpu_infeed_layout.cc
#include "xla/shape.h" #include "xla/stream_executor/tpu/c_api_conversions.h" #include "xla/stream_executor/tpu/tpu_api.h" #include "xla/translate/mhlo_to_hlo/type_to_shape.h" namespace mlir { static FailureOr<std::vector<int64_t>> GetTPUInfeedLayoutFromAPI( RankedTensorType t) { // Call the TPU API to determine the right infeed layout. Note that // this can fail if we're not running on a TPU-enabled node.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/test_base.h
// representation of a `ModuleOp` containing a single function `@main`. // Returns `failure` iff there is no `@main` or no such operation is found in // `@main`. template <typename OpT> FailureOr<OpT> FindFirstOpFromMainFunc(ModuleOp module_op) { func::FuncOp main_func_op = FindMainFuncOp(module_op); if (main_func_op == nullptr) return failure(); auto ops = main_func_op.getOps<OpT>();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 04 07:19:09 UTC 2024 - 3.5K bytes - Viewed (0)