- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 31 for NamedAttribute (0.41 sec)
-
tensorflow/compiler/mlir/tf2xla/api/v2/tf_executor_to_graph.cc
if (auto device_attr = func.getArgAttrOfType<mlir::StringAttr>(index, kDeviceAttr)) *node_def->mutable_device() = device_attr.getValue().str(); llvm::ArrayRef<mlir::NamedAttribute> func_arg_i_attrs = mlir::function_interface_impl::getArgAttrs(func, index); absl::flat_hash_set<absl::string_view> attrs_to_ignore = {kDeviceAttr,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Jun 06 23:04:51 UTC 2024 - 35.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/xla_call_module_deserialization.cc
<< " but enclosing XlaCallModule has a function list of size " << function_list.size(); return WalkResult::interrupt(); } llvm::SmallVector<NamedAttribute> new_config; // Copy the attributes in the current config except `called_index`. for (auto attr : backend_config) { if (attr.getName() != kCalledIndexAttrName) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 23 09:05:47 UTC 2024 - 11.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_device_propagation.cc
const llvm::DenseMap<Value, llvm::StringRef>& value_to_device) { // We apply all result attributes at once to avoid excessive allocations when // we have many result values. llvm::SmallVector<std::vector<NamedAttribute>, 8> result_attrs; { llvm::SmallVector<DictionaryAttr, 8> tmp; func.getAllResultAttrs(tmp); for (const auto& res : tmp) { result_attrs.push_back(res.getValue().vec()); } }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/bridge/convert_tf_quant_ops_to_mhlo.cc
rewriter.getIntegerType(64)), padding_nums); return padding_attr; } template <typename UniformQuantizedConvolutionOp> FailureOr<SmallVector<NamedAttribute>> ConvertToMhloConvolutionOpAttrs( UniformQuantizedConvolutionOp op, PatternRewriter &rewriter) { // TODO(b/261005147): Update the lowering logic after migration to mhlo // ConvolutionDimensionNumbers.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 30.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_dynamic_layout_pass.cc
llvm::ArrayRef<Type>{RankedTensorType::get({ShapedType::kDynamic}, builder->getIntegerType(64))}, llvm::ArrayRef<Value>{compilation_key}, llvm::ArrayRef<NamedAttribute>{ builder->getNamedAttr("index", builder->getI64IntegerAttr(execute_arg_index)), builder->getNamedAttr("is_output", builder->getBoolAttr(false))}); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 12.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/mlir_graph_optimization_pass_test.cc
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_; }; FunctionDef XTimesTwo() {
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/tensorflow/passes/insert_custom_aggregation_ops.cc
for (auto [value, aggregator_id] : llvm::zip_equal(quantizable_values, aggregator_ids)) { // ID attribute will have empty value for now. SmallVector<NamedAttribute, 5> attributes{ rewriter.getNamedAttr("id", rewriter.getStringAttr(aggregator_id)), rewriter.getNamedAttr( "calibration_method",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 14.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalize_tf_communication.cc
auto rendezvous_name = StringAttr::get(context, formatted_key); auto rendezvous_name_attr = NamedAttribute( StringAttr::get(context, xla::kXlaHostTransferRendezvousNameAttr), rendezvous_name); auto host_handler_name_value = StringAttr::get(context, host_handler_name.str()); auto host_handler_name_attr = NamedAttribute( StringAttr::get(context, xla::kXlaHostTransferHandlerNameAttr),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 40.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_device.cc
std::optional<DictionaryAttr> devices_attr; if (!devices.empty()) { llvm::SmallVector<mlir::NamedAttribute, 1> device_list; device_list.reserve(devices.size()); for (auto alias_and_devices : devices) { NamedAttribute device_name_attr = builder.getNamedAttr( alias_and_devices.getFirst(), builder.getStrArrayAttr(alias_and_devices.getSecond()));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 33.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_operator.cc
} Status mlir::CustomOptionsToAttributes( const std::string& custom_code, const std::vector<uint8_t>& custom_options, mlir::Builder builder, mlir::Location loc, llvm::SmallVectorImpl<mlir::NamedAttribute>* attributes) { attributes->emplace_back( builder.getNamedAttr("custom_code", builder.getStringAttr(custom_code))); std::string content; content.assign(reinterpret_cast<const char*>(custom_options.data()),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 38K bytes - Viewed (0)