- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 80 for created (0.09 sec)
-
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) -
tensorflow/compiler/mlir/quantization/stablehlo/cc/config.cc
config.mutable_calibration_options() ->mutable_representative_datasets() ->Add(preset_datasets.begin(), preset_datasets.end()); } // Create a new `QuantizationSpecs` to replace the existing one. The // expansion from `StaticRangePtqPreset` gets populated first and then // user-provided explicit `QuantizationSpec`s will be appended. QuantizationSpecs new_specs{};
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 03:36:50 UTC 2024 - 8.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/ir/UniformSupport.cc
return convert(attr); } return nullptr; } DenseElementsAttr UniformQuantizedPerAxisValueConverter::convert( DenseFPElementsAttr attr) { // Creates the converter for each chunk. Normally the size of the // quantization dim is 3, so we can cache all the converters. ShapedType type = attr.getType(); std::size_t dim_size = type.getDimSize(quantization_dim_);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 08 02:10:16 UTC 2024 - 4.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/reduce_type_precision.cc
dyn_cast<arith::ConstantOp>(op.getOperand(0).getDefiningOp()); if (!input_op) { return failure(); } Builder builder(op.getContext()); auto new_gather_op = rewriter.create<TFL::GatherOp>( op.getLoc(), /*result=*/ mlir::cast<TensorType>(op.getResult().getType()) .clone(builder.getI4Type()), /*operand=*/op.getOperands(), op->getAttrs());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/set_tpu_infeed_layout.cc
} else { /* If we're not running on a TPU node, we might not be able to * actually call the part of the TPU API that gives us layout. * This happens e.g. for unit tests. Below we just create a reasonable * layout. We sort by dimension size, which makes the layout agree with * the "correct" TPU layout in surprisingly many cases. * Note that the corresponding InfeedEnqueue op will be generated
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.1K bytes - Viewed (0)