- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 26 for iattr (0.22 sec)
-
tensorflow/compiler/jit/extract_outside_compilation_pass.cc
// we cannot rely on node name; use an attribute instead. host_compute_builder.Attr(kXlaOriginalOutsideCompilationNodeName, host_compute_builder.node_name()); // Copy all attributes. for (const auto& attr : call_node->attrs()) { host_compute_builder.Attr(attr.first, attr.second); } // Populate tpu_core assignment.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 104.7K bytes - Viewed (0) -
tensorflow/c/c_api.cc
TF_Status* status) { const tensorflow::AttrValue* attr = oper->node.attrs().Find(attr_name); if (attr == nullptr) { status->status = InvalidArgument("Operation '", oper->node.name(), "' has no attr named '", attr_name, "'."); } return attr; } TensorId ToTensorId(const TF_Output& output) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon Apr 15 03:35:10 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_export.cc
flex_builder->Bool(name.c_str(), attr.cast<mlir::vhlo::BooleanV1Attr>().getValue()); else if (llvm::isa<mlir::vhlo::StringV1Attr>(attr)) flex_builder->String( name.c_str(), attr.cast<mlir::vhlo::StringV1Attr>().getValue().str()); else if (llvm::isa<mlir::vhlo::IntegerV1Attr>(attr)) flex_builder->Int( name.c_str(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jun 12 21:41:49 UTC 2024 - 164.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
RecordValue(ValuePort(std::get<0>(result)), attr); } else { DCOMMENT("\t\tValue result unmapped, consider value type:" << value); RefineResultType(op, std::get<0>(result), value.getType()); } } if (ElementsAttr eattr = mlir::dyn_cast_or_null<ElementsAttr>(attr)) { if (std::get<0>(result).getType() == eattr.getType()) continue;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/legalize_hlo.cc
// According to the definition, the default value of these attributes // are all ones when unspecified. if (!attr.has_value()) return true; if (attr->getType().getShape()[0] != operand_type.getRank()) return false; if (!attr->isSplat()) return false; if (attr->getElementType() != rewriter.getIntegerType(64))
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 154.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_a_m.cc
DenseIntElementsAttr attr; if (axis_type.getElementType().isInteger(32)) { attr = DenseIntElementsAttr::get(axis_type, static_cast<int32_t>(axis)); } else { assert(axis_type.getElementType().isInteger(64)); attr = DenseIntElementsAttr::get(axis_type, axis); } auto axis_const = rewriter.create<TF::ConstOp>(loc, attr); auto concat =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 146.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
for (const auto& item : value.list().func()) { TF_ASSIGN_OR_RETURN(auto attr, ConvertFunctionCallName(item.name())); if (item.attr_size() != 0) return errors::Unimplemented( "func attributes with non-zero attr.size()"); if (attr) attrs.push_back(attr); } return builder_.getArrayAttr(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/optimize.cc
// Attribute holding a single value of float type. If attr has no elements, the // result is 0.0f. TypedAttr ConvertSingleElementAttrToFloatAttr(Attribute attr) { const auto dense_fp_attr = mlir::dyn_cast_or_null<DenseFPElementsAttr>(attr); if (dense_fp_attr) { // Already float => return return dense_fp_attr; } OpBuilder builder(attr.getContext());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 00:40:15 UTC 2024 - 102.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
auto check = [&](mlir::Value val, std::string attr_name) -> LogicalResult { ElementsAttr attr; if (matchPattern(val, m_Constant(&attr))) { if (attr.getShapedType().getRank() != 1) { return op.emitOpError() << "expects the rank of " << attr_name << "to be 1, got " << attr.getShapedType().getRank(); } if (input_ty.hasRank()) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0) -
tensorflow/compiler/jit/encapsulate_subgraphs_pass_test.cc
return false; } if (!EqualProtoMap<string, AttrValue>( a.attr(), b.attr(), [](const string& s) { return s; }, [](const AttrValue& v) { return v.DebugString(); }, [](const string& key, const AttrValue& av, const AttrValue& bv) { return av.DebugString() == bv.DebugString(); }, absl::StrCat("attr mismatch for function ", a.signature().name()), diff)) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Feb 22 08:47:20 UTC 2024 - 113.3K bytes - Viewed (0)