- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for StringAttr (0.17 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc
Operation* op = val.getDefiningOp(); // val should always have a defining op because cluster inputs always have // defining ops. assert(op); StringAttr sharding_attr = op->getAttrOfType<StringAttr>("_XlaSharding"); if (!sharding_attr) return context_op->emitOpError() << "A map_outside_compilation op's input should have an explicit "
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
// 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/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
} // Helper method that returns list of all strings in a StringAttr identified // by 'attr_key' and values are separated by a comma. llvm::SmallVector<llvm::StringRef, 2> GetStringsFromAttrWithSeparator( mlir::DictionaryAttr attr, const std::string& attr_key) { llvm::SmallVector<llvm::StringRef, 2> result; if (auto str = attr.get(attr_key).dyn_cast_or_null<mlir::StringAttr>()) { str.getValue().split(result, ',', /*MaxSplit=*/-1,
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/lite/transforms/prepare_tf.cc
}; // Transient state for preserving data from match to rewrite struct ConvertTFConvOpMatchState { IntegerAttr dilation_height_factor; IntegerAttr dilation_width_factor; StringAttr padding; IntegerAttr stride_height; IntegerAttr stride_width; }; // Templated class for declaring a converter from some TensorFlow convolution // op into its counterpart in TensorFlow Lite. //
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/tensorflow/transforms/lower_tf.cc
} // Returns reduction indices to use while lowering tf.BiasAddGrad op to tf.Sum // op. DenseIntElementsAttr GetBiasAddGradReductionIndices(int64_t rank, StringAttr data_format, Builder *builder) { tensorflow::TensorFormat format; if (!FormatFromString(data_format.getValue().str(), &format)) return {};
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 74.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/resource_op_lifting.cc
auto module = callee->getParentOfType<ModuleOp>(); name_base += "_resource_lifted"; auto name = name_base; callee = callee.clone(); callee.setPrivate(); callee.setName(mlir::StringAttr::get(callee->getContext(), name)); SymbolTable(module).insert(callee); result->lifted_callee = callee; // Remove unused resources in functions.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 55.1K bytes - Viewed (0)