- Sort Score
- Result 10 results
- Languages All
Results 101 - 110 of 168 for setFAttr (0.21 sec)
-
tensorflow/compiler/mlir/tensorflow/translate/tf_mlir_translate.cc
continue; } auto new_attr = mlir::DenseElementsAttr::get( llvm::cast<mlir::ShapedType>(attr.getType()), rand_val); inst.setAttr(attr_id, new_attr); } } } } return module_or; } absl::StatusOr<mlir::OwningOpRef<mlir::ModuleOp>> GraphdefToSplattedMlirTranslateFunction(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 07 11:51:44 UTC 2024 - 14.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/mlir_graph_optimization_pass_test.cc
auto min_consumer = b.getNamedAttr("min_consumer", b.getI32IntegerAttr(0)); auto bad_consumers = b.getNamedAttr("bad_consumers", b.getI32ArrayAttr({1, 2, 3, 4})); module->setAttr("tf.versions", b.getDictionaryAttr(llvm::ArrayRef<mlir::NamedAttribute>( {producer, min_consumer, bad_consumers}))); return run_status_; } Status run_status_;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Feb 27 08:25:30 UTC 2024 - 16.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/common/quantization_lib/quantization_utils.h
} rewriter.setInsertionPoint(quantizing_op); Operation* float_op = rewriter.clone(*quantizing_op); quantizing_op->setAttr(kDebugModeOpQuantAttrName, rewriter.getUnitAttr()); float_op->setAttr(kDebugModeOpFloatAttrName, rewriter.getUnitAttr()); RewireFloatModelBackbone(quantizing_op, float_op); return success(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 20:30:06 UTC 2024 - 41.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/xla_sharding_util.cc
llvm::SmallVector<mlir::Type, 4> output_types(num_split, output_type); *split_op = builder->create<mlir::TF::SplitOp>( location, output_types, split_dimension_op.getOutput(), src_input); (*split_op)->setAttr( kNumSplitAttr, builder->getIntegerAttr(builder->getIntegerType(32), num_split)); return mlir::success(); } // Creates a tf::ConcatOp that merges `input` values in `concat_dimension`.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 22 21:28:13 UTC 2024 - 34K 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/tensorflow/transforms/cluster_tf_ops_pass.cc
for (int i : llvm::seq<int>(0, metadata.result_devices.size())) { func_op.setResultAttr( i, kTFDeviceAttr, StringAttr::get(context, metadata.result_devices[i])); } func_op->setAttr(kHostAttr, StringAttr::get(context, host)); func_op.setPublic(); Block *block = func_op.addEntryBlock(); // Clones and moves the operations into the function's body. And the cloned
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 13.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_collective.cc
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(); } return op->emitOpError() << "module already contains an attribute "
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 16K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_device_inference.cc
if (auto device = result->DeviceForResource(output)) { LLVM_DEBUG(llvm::dbgs() << " Setting device = " << *device << "\n"); identity->setAttr(kDeviceAttr, builder.getStringAttr(*device)); } } } else if (auto while_region = dyn_cast<WhileRegionOp>(op)) { // For WhileRegion, do local analysis prior to visiting the attached
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 03 03:47:00 UTC 2023 - 13.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/quantization/tensorflow/fallback_to_flex_ops.cc
op->replaceAllUsesWith(flex_op); op->erase(); return true; } // Sets the "no_fallback" attribute. Value SetNoFallbackAttr(PatternRewriter &rewriter, Value val) { val.getDefiningOp()->setAttr(kNoFallbackAttr, rewriter.getUnitAttr()); return val; } // Returns true if the attr is a float attribute and be equal to value. static bool FloatValueEquals(const Attribute &attr, double value) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops.cc
// LINT.ThenChange(../../../../core/common_runtime/inline_function_utils.cc) for (Block &block : inlinedBlocks) { block.walk([&](Operation *op) { if (op->getDialect() == call->getDialect()) op->setAttr(device_id, node_device(op)); }); } } struct TFInlinerInterface : public DialectInlinerInterface { using DialectInlinerInterface::DialectInlinerInterface;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.6K bytes - Viewed (0)