- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 129 for created (0.11 sec)
-
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_save_op.cc
return builder.create<TF::ConstOp>( loc, DenseStringElementsAttr::get( tensor_type, SmallVector<StringRef>(str_values.begin(), str_values.end()))); } // Creates a 1D string array constant for "tensor_names" input of `RestoreV2` // op. The `ConstOp` will be created at `builder`'s current insertion point.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 05:52:39 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_restore_op.cc
return; } std::vector<TF::VarHandleOp> target_var_handle_ops = CollectVariableOps(session_init_func); if (target_var_handle_ops.empty()) { LOG(INFO) << "There are no VarHandleOps to restore. RestoreV2 op will not " "be created."; return; } CreateRestoreV2Op(target_var_handle_ops, session_init_func); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Mar 12 06:02:20 UTC 2023 - 9K bytes - Viewed (0) -
tensorflow/cc/ops/while_loop.cc
// before the NextIteration node is created. string NextIterationName(const Scope& scope, int loop_var_idx) { string result; const string& prefix = scope.impl()->name(); if (!prefix.empty()) strings::StrAppend(&result, prefix, "/"); strings::StrAppend(&result, "NextIteration"); if (loop_var_idx > 0) strings::StrAppend(&result, "_", loop_var_idx); return result; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 01:01:21 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/tf_tfl_translate_cl.cc
llvm::cl::init("")); // A list of comma separated TF operators which are created by the user. // This must be used with `-emit-select-tf-ops=true`. // NOLINTNEXTLINE opt<std::string> select_user_tf_ops( "select-user-tf-ops", llvm::cl::desc( "<list of custom tf ops created by the user (comma separated)>"), llvm::cl::init("")); // NOLINTNEXTLINE
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 05 20:53:17 UTC 2024 - 7.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/decompose_resource_ops_pass.cc
"Resource ops decomposition did not converge"; // Decomposing resource ops should not take more than a few iterations (2-3) to // converge as only a few patterns create new resource ops that can be further // decomposed. The rest of the iterations are enough to clean up any dead ops // created by decomposition. constexpr int kMaxIterations = 20; // Populates `reachable_functions` with all functions that can be reached from // device cluster ops.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Sep 08 20:01:13 UTC 2023 - 8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/python/unfreeze_constants.cc
pm.addPass(mlir::quant::CreateUnfreezeConstantsPass()); }, ctx, module_op)); if (const absl::Status create_dir_status = Env::Default()->CreateDir(std::string(checkpoint_dir)); !create_dir_status.ok()) { LOG(ERROR) << "Failed to create checkpoint directory at: " << checkpoint_dir; return create_dir_status; } TF_ASSIGN_OR_RETURN(const auto unused_variable_names,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sun Apr 14 11:07:03 UTC 2024 - 3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_layout_helper.cc
return {0, 2, 3, 1}; } else { return {}; } } // Shuffle elements in the `attr` according to the permutation. Optional // `inner_size` allows to shuffle array attributes created from rank 2 tensors // on outer dimension only. ArrayAttr ShuffleArrayAttr(ArrayAttr attr, ArrayRef<int64_t> permutation, int inner_size) { if (attr.empty()) return attr;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 3.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_host_computation_expansion.cc
constexpr char kXlaOutsideCompilationAttr[] = "_xla_outside_compilation"; bool HasOutsideCompilationAttribute(Operation* op) { return op->getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr) != nullptr; } // Finds op that created a given value. If the value is a BlockArgument, this // returns the owner of the Block. Operation* GetOpOfValue(Value value) { if (auto block_arg = mlir::dyn_cast<BlockArgument>(value))
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/tensorflow/debugging/mlir_dump.cc
const std::string dump_file_path = tsl::io::JoinPath(*dump_dir, dump_file_name); TF_ASSIGN_OR_RETURN(std::unique_ptr<llvm::raw_ostream> file, WritableFileWrapper::Create(dump_file_path)); LOG(INFO) << "IR dump file created: " << dump_file_path; return file; } class PrinterConfig : public mlir::PassManager::IRPrinterConfig { public: explicit PrinterConfig(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 31 05:38:57 UTC 2024 - 9.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/quantization_unit_loc.cc
// If the op is rewritten, FusedLoc can be created. for (Location child_loc : mlir::cast<FusedLoc>(loc).getLocations()) { std::optional<QuantizationUnitLoc::QuantizationUnit> found_unit = FindQuantizationUnitFromLoc(child_loc); if (found_unit.has_value()) return found_unit; } } else if (isa<CallSiteLoc>(loc)) { // If the graph is inlined, CallSiteLoc can be created. return FindQuantizationUnitFromLoc(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 4.2K bytes - Viewed (0)