- Sort Score
- Result 10 results
- Languages All
Results 11 - 16 of 16 for AttrSlice (0.31 sec)
-
tensorflow/cc/framework/scope.cc
std::unordered_set<string> Scope::Impl::GetColocationConstraints( const Operation& colocate_with_op) const { std::unordered_set<string> current_constraints(colocation_constraints_); const AttrSlice attrs = colocate_with_op.node()->attrs(); std::vector<string> node_constraints; if (TryGetNodeAttr(attrs, kColocationAttrName, &node_constraints)) { for (const string& entry : node_constraints) {
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/kernels.cc
TF_Status* status) { auto* cc_ctx = reinterpret_cast<::tensorflow::OpKernelConstruction*>(ctx); const tensorflow::AttrValue* attr = ::tensorflow::AttrSlice(cc_ctx->def()).Find(attr_name); if (attr == nullptr) { status->status = absl::InvalidArgumentError( absl::StrCat("Operation '", cc_ctx->def().name(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 22:53:47 UTC 2024 - 36K bytes - Viewed (0) -
tensorflow/compiler/jit/kernels/xla_ops.cc
has_ref_vars_(has_ref_vars) {} void XlaLocalLaunchBase::ComputeAsync(OpKernelContext* ctx, DoneCallback done) { VLOG(1) << "XlaLocalLaunchOpBase::Compute " << Canonicalize(function_.name(), AttrSlice(&function_.attr())); xla_launch_counter->GetCell(platform_info_.device_type().type_string()) ->IncrementBy(1); std::vector<const Tensor*> inputs = InputsFromContext(ctx);
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/compiler/jit/mark_for_compilation_pass_test.cc
{"dx: float"}, // Attr defs {}, // Nodes nodes); return absl::OkStatus(); } // A gradient containing only supported operators Status SupportedGrad(const AttrSlice& attrs, FunctionDef* g) { // clang-format off return GradForUnaryCwise(g, { {{"y"}, "Tanh", {"x"}}, {{"y2"}, "Square", {"y"}, {}, {"dy"}}, FunctionDefHelper::Const("one", 1.0f),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 14 10:11:10 UTC 2024 - 79.6K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass.cc
call_node_->set_assigned_device_name(device_); return absl::OkStatus(); } Status Encapsulator::GetFunctionNameAttr(Node const* node, string* attr) const { AttrSlice attrs = node->attrs(); attr->clear(); for (const auto& node_attr : attrs) { if (node_attr.first == group_attribute_) { TF_RETURN_IF_ERROR(AttrValueHasType(node_attr.second, "string"));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 51K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
conversion_metadata.function_name); std::unique_ptr<FunctionBody> fbody; TF_RETURN_IF_ERROR( FunctionDefToBodyHelper(*func_def, AttrSlice(), &graph_flib_, &fbody)); TF_RETURN_IF_ERROR(importer.PrepareConvert(*fbody->graph)); TF_ASSIGN_OR_RETURN(auto func_type, importer.InferLibFunctionType(*fbody));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0)