- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 97 for attr_value (0.24 sec)
-
tensorflow/c/eager/tfe_op_attrs_internal.h
typedef struct TFE_Op TFE_Op; namespace tensorflow { DEFINE_CONVERSION_FUNCTIONS(tensorflow::AbstractOpAttrs, TFE_OpAttrs); // Set an AttrValue on the op. Doesn't handle the list types. void SetOpAttrValueScalar(TFE_Context* ctx, TFE_Op* op, const tensorflow::AttrValue& default_value, const char* attr_name, TF_Status* status); } // namespace tensorflow
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Feb 10 05:41:19 UTC 2021 - 1.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_layout_helper.h
(*op)->setAttr("data_format", StringAttr::get(context, target_data_format)); for (auto pair : shuffle_attrs) { StringRef attr_name = pair.first; ArrayAttr attr_value = pair.second; (*op)->setAttr(attr_name, ShuffleArrayAttr(attr_value, reverse_permutation)); } auto fold = cast<FoldOperandsTransposeInterface>(op->getOperation()); for (unsigned idx : fold.GetLayoutDependentResults()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 08 01:19:25 UTC 2023 - 5.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_collective.cc
IntegerAttr attr_value, Operation* op, ModuleOp& module) { const auto ex_attr_value = module->getAttrOfType<IntegerAttr>(attr_name); if (ex_attr_value == nullptr) { module->setAttr(attr_name, attr_value); return success(); } if (ex_attr_value == attr_value) { return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16K bytes - Viewed (0) -
tensorflow/c/c_test_util.h
bool GetFunctionDef(TF_Function* func, tensorflow::FunctionDef* func_def); bool GetAttrValue(TF_Operation* oper, const char* attr_name, tensorflow::AttrValue* attr_value, TF_Status* s); // Returns a sorted vector of std::pair<function_name, gradient_func> from // graph_def.library().gradient() std::vector<std::pair<string, string>> GetGradDefs(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Aug 09 01:06:53 UTC 2018 - 6K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_util.cc
Status AppendToListAttr(Node* n, const string& attr_name, const string& value) { std::vector<T> attr_value; Status s = GetNodeAttr(n->attrs(), attr_name, &attr_value); if (!s.ok() && s.code() != error::NOT_FOUND) { return s; } n->ClearAttr(attr_name); attr_value.push_back(value); n->AddAttr(attr_name, attr_value); return absl::OkStatus(); } // Replaces attribute value. template <typename T>
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/tf2xla/transforms/legalize_tf_with_tf2xla.cc
} } return true; } bool HasSymbolRefAttr(Operation* op) { for (const auto& attr : op->getAttrs()) { Attribute attr_value = attr.getValue(); if (mlir::isa<SymbolRefAttr>(attr_value)) { return true; } else if (auto array_attr = mlir::dyn_cast<ArrayAttr>(attr_value)) { if (!array_attr.empty() && mlir::isa<SymbolRefAttr>(*array_attr.begin())) { return true; } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 9.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/export_tf_dialect_op.h
#include "llvm/ADT/StringRef.h" #include "mlir/IR/Operation.h" // from @llvm-project #include "tensorflow/compiler/mlir/tensorflow/utils/export_utils.h" #include "tensorflow/core/framework/attr_value.pb.h" #include "tensorflow/core/framework/node_def.pb.h" #include "tensorflow/core/framework/node_def_util.h" #include "tensorflow/core/framework/op_def_builder.h" #include "tensorflow/core/platform/status.h"
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 2.4K bytes - Viewed (0) -
tensorflow/c/c_api_function.cc
const void* proto, size_t proto_len, TF_Status* status) { tensorflow::AttrValue attr_value; if (!attr_value.ParseFromArray(proto, proto_len)) { status->status = InvalidArgument( "Unparseable AttrValue proto passed to " "TF_FunctionSetAttrValueProto"); return; } auto fdef_or = func->record->mutable_fdef();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 13.6K bytes - Viewed (0) -
tensorflow/c/eager/abstract_op_attrs.h
#ifndef TENSORFLOW_C_EAGER_ABSTRACT_OP_ATTRS_H_ #define TENSORFLOW_C_EAGER_ABSTRACT_OP_ATTRS_H_ #include "absl/container/inlined_vector.h" #include "tensorflow/core/framework/attr_value.pb.h" #include "tensorflow/core/framework/types.pb.h" #include "tensorflow/core/platform/status.h" namespace tensorflow { // Attributes of an op. class AbstractOpAttrs { protected:
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 26 22:20:27 UTC 2021 - 2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/utils/perception_ops_utils_test.cc
mlir::NameLoc::get(builder->getStringAttr("fused_func")), "fused_func", func_type, {}); func.addEntryBlock(); mlir::StringAttr attr_value = builder->getStringAttr("MaxUnpooling2D"); func->setAttr("tf._implements", attr_value); return func; } func::FuncOp createMaxUnpoolingFunc( mlir::Builder* builder, const SmallVector<int64_t, 4>& input_shape,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Sep 29 21:02:21 UTC 2022 - 7.4K bytes - Viewed (0)