- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 58 for created (0.11 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/add_quantization_unit_loc.cc
return isa<NameLoc>(loc) || (isa<CallSiteLoc>(loc) && isa<NameLoc>(mlir::cast<CallSiteLoc>(loc).getCallee())); }); } // Finds the pattern of the location created by `ImporterBase::GetLocation` // in `tensorflow/compiler/mlir/tensorflow/translate/import_model.cc`. void FindQuantizationUnitsRecursively(Location loc, SmallVector<QuantizationUnit>& units) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/lift_variables.cc
// Create tensor attribute for this variable. absl::StatusOr<ElementsAttr> tensor_attr_or = ConvertTensor(tensor, &builder); if (!tensor_attr_or.ok()) { return module.emitOpError() << "failed to convert tensor (name: " << name.str() << ")"; } ElementsAttr tensor_attr = tensor_attr_or.value(); builder.create<tf_saved_model::GlobalTensorOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 7.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/tensorflow/tf_to_quant.cc
// Finally, use the quantization parameter to create the quantize and // dequantize ops, and insert them between the tf.FakeQuantWithMinMaxVarsOp // and its users. Value value = tf_op.getOutputs(); auto quantize = rewriter.create<quantfork::QuantizeCastOp>( tf_op.getLoc(), qtype.getValue(), value); auto dequantize = rewriter.create<quantfork::DequantizeCastOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tf_saved_model_asset_sinking_pass.cc
// Create SessionInitializerOp; should reference main function. StringAttr func_name = main_func.getSymNameAttr(); llvm::SmallVector<mlir::Attribute, 2> func_names = { mlir::SymbolRefAttr::get(builder.getContext(), func_name)}; builder.create<tf_saved_model::SessionInitializerOp>( module->getLoc(), builder.getArrayAttr(func_names)); // Create AssetOp; this holds the checkpoint_path.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/io.cc
namespace stablehlo::quantization::io { absl::StatusOr<std::string> GetLocalTmpFileName(tsl::Env* const env) { std::string tmp_fname{}; if (!env->LocalTempFilename(&tmp_fname)) { return absl::InternalError("Failed to create tmp file name."); } return tmp_fname; } absl::StatusOr<std::string> GetLocalTmpFileName() { return GetLocalTmpFileName(tsl::Env::Default()); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 03:28:15 UTC 2024 - 2.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/io_test.cc
// A test-only derived class of `tsl::Env` which is broken. Used to cause // failure for the `CreateTmpDir` function. Each of the overridden member // functions implements a dummy functionality just to be able to create an // instance of this class. class TestEnvBrokenFileSystem : public tsl::Env { public: TestEnvBrokenFileSystem() = default; bool MatchPath(const tsl::string& path, const tsl::string& pattern) override {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 03:28:15 UTC 2024 - 6.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/experimental/common/outline_operations.cc
FunctionType function_type = builder.getFunctionType(input_types, return_types); std::string function_name = absl::StrCat("func_", subgraph.subgraph_id_); func::FuncOp new_func = func::FuncOp::create(builder.getUnknownLoc(), function_name, function_type); new_func.setVisibility(func::FuncOp::Visibility::Private); new_func.addEntryBlock();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 8.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/convert_asset_args_test.cc
ConvertAssetArgsTest() { ctx_.loadDialect<func::FuncDialect, TF::TensorFlowDialect, tf_saved_model::TensorFlowSavedModelDialect>(); } // Parses `module_op_str` to create a `ModuleOp`. Checks whether the created // module op is valid. OwningOpRef<ModuleOp> ParseModuleOpString( const absl::string_view module_op_str) { auto module_op_ref = parseSourceString<ModuleOp>(module_op_str, &ctx_);
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/tensorflow/transforms/initialize_variables_in_session_init.cc
auto var_handle_op_in_init = var_handle_op->clone(); builder.insert(var_handle_op_in_init); auto const_op = builder.create<mlir::arith::ConstantOp>( session_init_func.getLoc(), tensor_attr.getType(), tensor_attr); builder.create<TF::AssignVariableOp>( session_init_func.getLoc(), llvm::ArrayRef<mlir::Type>{}, llvm::ArrayRef<mlir::Value>{var_handle_op_in_init->getResult(0),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_partitioned_op_conversion.cc
} if constexpr (is_input) { auto pi = builder.create<TF::TPUPartitionedInputV2Op>( op.getLoc(), op.getType(), op.getOperands(), builder.getI64ArrayAttr(partition_dims), builder.getBoolAttr(false), op.get_XlaShardingAttr()); op->replaceAllUsesWith(pi); } else { auto po = builder.create<TF::TPUPartitionedOutputV2Op>( op.getLoc(), op.getResultTypes(), op.getOperand(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.1K bytes - Viewed (0)