- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 43 for Dadd (0.03 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/cluster_ops_by_policy.cc
// operation dominates all the users (%3 in this example). Operation *insertion_point; // After construction: // A set of constraints on the clustered operation operands that must be // satisfied in order to add operation to the cluster. For basic block // source this will be always empty. // // During the union-find cluster formation: // The root member will have constraints merged from all of the cluster
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_variable_runtime_reformatting.cc
devices.insert({device_alias, device_list_for_alias}); } OpBuilder builder(replicate); builder.setInsertionPoint(while_op); // Create per-device variables for formatting state, and add them to the while // loop. auto key_type = RankedTensorType::get({2}, TF::StringType::get(builder.getContext())); auto state_vars = CreateStateVars(devices, while_op.getLoc(), key_type, &builder);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/cc/save_variables_test.cc
%add = "tf.AddV2"(%cst, %cst) : (tensor<2xf32>, tensor<2xf32>) -> tensor<2xf32> %var_handle = "tf.VarHandleOp"() {container = "", device = "/device:CPU:0", shared_name = "var_0"} : () -> tensor<!tf_type.resource<tensor<2xf32>>> "tf.AssignVariableOp"(%var_handle, %add) : (tensor<!tf_type.resource<tensor<2xf32>>>, tensor<2xf32>) -> () return }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 16.2K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_util.cc
edges.push_back(EdgeInfo{e->dst_input(), e->dst()->id()}); VLOG(4) << "Oc -> oc edge: " << e->DebugString(); } } // Remove the edge from host to outside compilation. Add a placeholder as // outside compilation node input. std::map<std::pair<string, int>, Node*> placeholders; for (int i = 0, end = edges.size(); i < end; i++) { Node* dst = g->FindNodeId(edges[i].dst_node_id);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 15.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_rewrite_pass.cc
auto compilation_status_type = RankedTensorType::get({}, builder->getType<TF::StringType>()); auto program_type = RankedTensorType::get({3}, builder->getType<TF::StringType>()); // Add MLIR module's fingerprint to compile metadata. uint64_t mlir_fingerprint = tensorflow::Fingerprint64(txt_module); metadata.set_mlir_fingerprint(mlir_fingerprint); std::string txt_metadata;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 29.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/tpu_sharding_identification_pass.cc
tensorflow::DeviceNameUtils::ParsedName name; if (tensorflow::DeviceNameUtils::ParseFullName(device.str(), &name)) { if (name.type == "TPU_REPLICATED_CORE") { // TODO(hanxiongwang): Add check for out of bound of name.id return llvm::StringRef(logical_device_vec[name.id]); } } } } return std::nullopt; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 02:01:13 UTC 2024 - 28.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/prepare_quantize_dynamic_range.cc
} else if (quant_specs_.inference_type == tensorflow::DT_HALF) { quantizeOpAsFloat16(rewriter, op, quant_op); quantized = true; } } return quantized; } // Add asymmetric input quantization attribute. MLIR dynamic quantization // supports only the case that the value of the attribute equals to true. For // details, see tensorflow/compiler/mlir/lite/quantization/quantization.td
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 20.8K bytes - Viewed (0) -
tensorflow/cc/framework/scope.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Apr 13 05:57:22 UTC 2024 - 20.9K bytes - Viewed (0) -
tensorflow/c/experimental/stream_executor/stream_executor.cc
if (memory_bandwidth >= 0) { builder.set_memory_bandwidth(memory_bandwidth); } } // TODO(annarev): Add gflops field in DeviceDescription and set it here. // TODO(annarev): Perhaps add `supports_unified_memory` in // DeviceDescription. return builder.Build(); } absl::StatusOr<std::unique_ptr<Event>> CreateEvent() override {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 14 07:39:19 UTC 2024 - 27.1K bytes - Viewed (0) -
tensorflow/cc/framework/cc_op_gen_util.cc
graph_op_def.deprecation().version(), ":\n", graph_op_def.deprecation().explanation(), ".\n"); } else if (api_def.summary().empty()) { comment = "TODO: add doc.\n"; } else { comment = strings::StrCat(api_def.summary(), "\n"); } if (!api_def.description().empty()) { strings::StrAppend(&comment, "\n", api_def.description(), "\n"); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Feb 26 00:57:05 UTC 2024 - 25K bytes - Viewed (0)