- Sort Score
- Result 10 results
- Languages All
Results 61 - 70 of 155 for created (0.15 sec)
-
tensorflow/compiler/jit/kernels/xla_ops.cc
compilation_successful.scalar<bool>()() = false; ctx->set_output(0, compilation_key); ctx->set_output(1, compilation_successful); return; } // Each execution of an XlaCompile op creates a new ExecutableClosure, even // if it didn't have to compile the cluster because of a compilation-cache // hit. This is because we at least need new snapshots of the resource // variables.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 22:46:36 UTC 2024 - 41.4K bytes - Viewed (0) -
tensorflow/cc/framework/gradients.cc
std::vector<Output>* grad_outputs); // Returns a list mapping whether each node in the graph is reachable // from outputs_. Keyed by node id. std::vector<bool> GetReachableNodes(); // Creates the gradient subgraph for a while loop (or just stores // `summed_grads` if not all incoming gradients are available yet). All exit // nodes (which are the first nodes of a loop encountered in the backwards
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 05:57:22 UTC 2024 - 22K bytes - Viewed (0) -
tensorflow/compiler/jit/xla_platform_info.cc
const std::string& profiler_name = GetPjRtDeviceCompilationProfilerResourceName(device_type); bool deleted_old_device_compiler = false; // Lookup the DeviceCompiler, create one if not found. Status s = rm->Lookup<PjRtDeviceCompiler>( rm->default_container(), compiler_name, pjrt_device_compiler); if (s.ok() && device_type == DEVICE_TPU) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 17:23:27 UTC 2024 - 17.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/tpu_rewrite_device_util.cc
for (const auto& device : devices) if (DeviceNameUtils::IsCompleteSpecification(spec, device)) matching_devices.push_back(device); return matching_devices; } // Create error message for a conflicting attribute of a device. template <typename T> absl::Status MismatchedTPUSystemAttributeErr(absl::string_view attribute, T a, T b) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Jun 10 20:10:40 UTC 2024 - 32.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_ops_by_policy.cc
return a->isBeforeInBlock(b); }); // Create tf_device::ClusterOp before the last operation in the block that // is a part of a match set. auto back = cluster.operations.back(); auto loc = back->getLoc(); OpBuilder builder(back); auto cluster_op = builder.create<tf_device::ClusterOp>(loc, return_types, policy); // Create block in cluster_op's region and move 'cluster.operations' into
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 27.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_metadata_utils.cc
"bad '{0}' attribute at index {1} with value '{2}': failed to parse to {3}"; constexpr char kBadArrayAttrLengthMsg[] = "bad '{0}' attribute, expected array attribute of size {1}, got size {2}"; // Creates a missing attribute error message. std::string CreateMissingAttributeMsg(llvm::StringRef attribute) { return llvm::formatv("requires attribute '{0}'", attribute).str(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_to_mhlo_int_test.cc
// can't lower tf.Const. Value cst; if (use_mhlo_const) { cst = builder.create<mhlo::ConstantOp>(func_op->getLoc(), attrs); } else { cst = builder.create<TF::ConstOp>(func_op->getLoc(), attrs); } func_op.getArgument(i).replaceAllUsesWith(cst); } return module_op; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 03 01:03:21 UTC 2024 - 35.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/lstm_utils_test.cc
projection_type, layer_norm_scale_type}; auto func_type = builder->getFunctionType(input_types, output_type); auto func = func::FuncOp::create( mlir::NameLoc::get(builder->getStringAttr("fused_func")), "fused_func", func_type, {}); func.addEntryBlock(); std::vector<std::string> attributes; if (ln) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 10K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/tfl_stablehlo_pass.cc
for (int i = 0; i < custom_op.getNumResults(); i++) { output_tys.push_back(custom_op.getType(i)); } op_state.addTypes(output_tys); op_state.addAttributes(attr); auto stablehlo_op = builder.create(op_state); custom_op.replaceAllUsesWith(stablehlo_op); custom_op.erase(); }); } std::unique_ptr<OperationPass<func::FuncOp>> CreateTflToStablehloPass() {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 24 06:08:43 UTC 2024 - 10.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_deserialization.cc
for (auto attr : op.getPlatforms().getAsRange<StringAttr>()) { platforms.push_back(attr.getValue().str()); } TF_ASSIGN_OR_RETURN( auto loader, tensorflow::XlaCallModuleLoader::Create( context, static_cast<int>(op.getVersion()), op.getModule().str(), std::move(disabled_checks), std::move(platforms), /*num_invocation_args=*/op.getArgs().size(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 11.1K bytes - Viewed (0)