- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for getAttrOfType (0.23 sec)
-
tensorflow/compiler/mlir/lite/transforms/prepare_tf.cc
auto bias = rewriter.create<TF::ConstOp>(op->getLoc(), bias_type, bias_attr); if (op->getAttrOfType<StringAttr>("padding").getValue() == "EXPLICIT") { // Add Const op for padding value. ArrayRef<Attribute> padding_attr_array = op->getAttrOfType<ArrayAttr>("explicit_paddings").getValue(); auto get_int = [](Attribute attr) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 28 21:49:50 UTC 2024 - 64.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/quantize_composite_functions.cc
// contain the attribute names. // TransferQuantizationAttributes(rewriter, inner_op, attrs); std::string attr_map_str = inner_op.getAttrOfType<StringAttr>(kAttrMapAttribute).str(); for (absl::string_view element_str : absl::StrSplit(attr_map_str, ',')) { std::vector<absl::string_view> key_and_value_pair = absl::StrSplit(element_str, ':');
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/tf2xla/internal/passes/extract_outside_compilation.cc
// outside compilation. LogicalResult CheckAncestorNotOutsideComp(Operation* op) { if (!op->getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) return mlir::success(); Operation* iter_op = op; while (auto* parent_op = iter_op->getParentOp()) { if (parent_op->getAttrOfType<StringAttr>(kXlaOutsideCompilationAttr)) { return op->emitOpError()
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/sparsecore/embedding_pipelining.cc
return true; } StringAttr GetReplicationAttr(mlir::Operation* op) { return op->getAttrOfType<StringAttr>(TF::kReplicationInfoAttr); } StringAttr GetReplicationAttr(TF::TPUCompilationResultOp op) { // Special case for getting the replication region for // TPUCompilationResultsOp. return op->getAttrOfType<StringAttr>(kTpuCompilationStatus); }
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
const std::vector<std::unique_ptr<tflite::TensorT>>& tensors) { auto* context = func->getContext(); static const char kEntryFunctionAttributes[] = "tf.entry_function"; auto dict_attr = func->getAttrOfType<mlir::DictionaryAttr>(kEntryFunctionAttributes); if (!dict_attr) return; // Get Input and output tensor names from attribute. llvm::SmallVector<llvm::StringRef, 2> input_names =
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 66.8K bytes - Viewed (0)