- Sort Score
- Result 10 results
- Languages All
Results 1 - 10 of 35 for getNamedAttr (0.18 sec)
-
tensorflow/compiler/mlir/lite/flatbuffer_operator.cc
attributes.emplace_back(builder.getNamedAttr( "backend_config", BuildVhloStringV1Attr(op->backend_config, builder))); attributes.emplace_back(builder.getNamedAttr( "called_computations", BuildVhloArrayV1Attr({}, builder))); attributes.emplace_back(builder.getNamedAttr( "operand_layouts", BuildVhloArrayV1Attr({}, builder))); attributes.emplace_back(builder.getNamedAttr(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue May 21 18:21:50 UTC 2024 - 38K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/utils/tf_to_uniform_attribute_utils.cc
} } attrs.push_back(rewriter.getNamedAttr("lhs_quantization_axis", activation_quantization_axis)); attrs.push_back(rewriter.getNamedAttr("rhs_quantization_axis", activation_quantization_axis)); attrs.push_back(rewriter.getNamedAttr("output_quantization_axis",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 18.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/stablehlo/transforms/tfl_stablehlo_pass.cc
const auto& value = m[key]; switch (value.GetType()) { case flexbuffers::FBT_INT: { auto attr = value.AsInt64(); auto named_attr = builder->getNamedAttr( key, builder->getIntegerAttr(builder->getIntegerType(64), attr)); attrs.push_back(named_attr); break; } case flexbuffers::FBT_VECTOR_BOOL: {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Jan 24 06:08:43 UTC 2024 - 10.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/readonly_references_to_resources.cc
ArrayRef<Value>{}, ArrayRef<NamedAttribute>{ builder.getNamedAttr("device", device_attr), builder.getNamedAttr("container", variable_v2_op.getContainerAttr()), builder.getNamedAttr("shared_name", builder.getStringAttr(variable_name))}); for (Operation *user :
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 7.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/translate_utils.cc
mlir::Builder b(module.getContext()); auto producer = b.getNamedAttr("producer", b.getI32IntegerAttr(versions.producer())); auto min_consumer = b.getNamedAttr( "min_consumer", b.getI32IntegerAttr(versions.min_consumer())); auto bad_consumers = b.getNamedAttr( "bad_consumers", b.getI32ArrayAttr(llvm::ArrayRef<int32_t>( versions.bad_consumers().data(),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 3.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/add_dump_tensor_op.cc
SmallVector<NamedAttribute> dump_attributes{ rewriter.getNamedAttr("log_dir_path", rewriter.getStringAttr(folder_name)), rewriter.getNamedAttr("file_name", rewriter.getStringAttr(file_name)), // The op is disabled by default. Otherwise, values will be saved // during calibration. rewriter.getNamedAttr("enabled", rewriter.getBoolAttr(enabled)),
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Mar 22 22:55:22 UTC 2024 - 13K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/insert_custom_aggregation_ops.cc
rewriter.getNamedAttr("id", rewriter.getStringAttr(aggregator_id)), rewriter.getNamedAttr( "calibration_method", rewriter.getI32IntegerAttr(calib_opts_.calibration_method())), rewriter.getNamedAttr("num_bins", rewriter.getI32IntegerAttr(effective_num_bins)), rewriter.getNamedAttr( "min_percentile",
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/tensorflow/utils/device_util_test.cc
mlir::ModuleOp::create(mlir::UnknownLoc::get(&context)); mlir::Builder builder(*module_ref); (*module_ref) ->setAttr("tf.devices", builder.getDictionaryAttr(builder.getNamedAttr( "bad_device", builder.getDictionaryAttr({})))); mlir::TF::RuntimeDevices devices; EXPECT_TRUE(mlir::failed(GetDevicesFromOp(*module_ref, &devices))); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 9.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfrt/ir/tfrt_fallback_common.cc
auto key = key_value.getName(); auto value = key_value.getValue(); op_attr_array.push_back(builder.getArrayAttr({key, value})); } result.attributes.push_back( builder.getNamedAttr("op_attrs", builder.getArrayAttr(op_attr_array))); // TODO(tfrt-devs): support func attributes in tfrt_fallback_sync. if (options.has_func_attr) { llvm::SmallVector<mlir::Attribute, 4> op_func_attr_array;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 5.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/device_util.cc
if (device->device_type() == DEVICE_GPU) { auto metadata = ParseGpuDeviceMetadata(*device, &builder); devices.push_back(builder.getNamedAttr(name, metadata)); } else { auto metadata = builder.getUnitAttr(); devices.push_back(builder.getNamedAttr(name, metadata)); } } op->setAttr(kDevicesAttr, builder.getDictionaryAttr(devices)); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 6.4K bytes - Viewed (0)