- Sort Score
- Result 10 results
- Languages All
Results 1 - 6 of 6 for SetAttr (0.12 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_pipelining.cc
for (const auto& attr_name : attr_names) { if (!global_iter_id_op->hasAttr(attr_name)) continue; offset->setAttr(attr_name, global_iter_id_op->getAttr(attr_name)); new_global_iter_id->setAttr(attr_name, global_iter_id_op->getAttr(attr_name)); } // Make the next function to get inlined use a different offset. ++offset_value; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 92.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_import.cc
// use it in exports or if there are breaking changes module->setAttr("tfl.schema_version", builder.getI32IntegerAttr(model->version)); if (!model->description.empty()) { module->setAttr("tfl.description", builder.getStringAttr(model->description)); } if (!metadata_attrs.empty()) { module->setAttr("tfl.metadata", builder.getDictionaryAttr(metadata_attrs)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 66.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc
Operation* parent_op = op->getParentOp(); if (!parent_op) return false; return HasOutsideCompilationAncestor(parent_op); } Operation* ApplyXlaHostTransferAttr(Operation* op, OpBuilder& builder) { op->setAttr("_xla_has_host_transfer", builder.getBoolAttr(true)); return op; } // Creates a tf._XlaSendFromHost or tf._XlaSendFromHostV2 op. If device ordinal // is present, a tf._XlaSendFromHostV2 op is created instead.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 68.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 54.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc
auto new_call = builder.create<CallOpType>( call_op.getLoc(), lifting_info.lifted_callee.getFunctionType().getResults(), new_operands, call_op->getAttrs()); new_call->setAttr("f", SymbolRefAttr::get(builder.getContext(), lifting_info.lifted_callee.getName())); AddLoadsStoresOutsideControlFlowOp(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 55.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
llvm::all_of(op->getResultTypes(), is_not_variant); }; auto is_set_item_legal = [](Operation *op) { return op->hasAttr("resize_if_index_out_of_bounds") && op->getAttr("resize_if_index_out_of_bounds") .cast<mlir::BoolAttr>() .getValue(); }; ConversionTarget target(*context);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0)