- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for getStringAttr (0.29 sec)
-
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_outside_compilation.cc
auto host_compute = builder.create<mlir::TF::_XlaHostComputeMlirOp>( loc, shard_output_types, manual_inputs, /*send_key=*/builder.getStringAttr(args_communication_key), /*recv_key=*/builder.getStringAttr(retvals_communication_key), /*host_mlir_module=*/builder.getStringAttr(serialized_func_module), /*manual_sharding=*/builder.getBoolAttr(true)); host_compute_out_ops.push_back(host_compute);
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/lite/flatbuffer_import.cc
metadata_attrs.emplace_back(builder.getStringAttr(metadata->name), builder.getStringAttr("true")); continue; } std::vector<uint8_t> buffer = model->buffers[metadata->buffer]->data; metadata_attrs.emplace_back( builder.getStringAttr(metadata->name), builder.getStringAttr(llvm::StringRef(
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/tensorflow/transforms/sparsecore/embedding_pipelining.cc
auto caller = builder.create<TF::StatefulPartitionedCallOp>( loc, result_types, operands, symbol, /*config=*/builder.getStringAttr(""), /*config_proto=*/builder.getStringAttr(""), /*executor_type=*/builder.getStringAttr("")); caller.setFAttr(symbol); // Set an attribute that our inliner will look for when choosing which // TF::StatefulPartitionedCallOps to inline.
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/transforms/prepare_tf.cc
target_values.push_back((*max.begin())); if (auto tmpAttr = num_bits) { target_attrs.emplace_back(rewriter.getStringAttr("num_bits"), tmpAttr); } if (auto tmpAttr = narrow_range) { target_attrs.emplace_back(rewriter.getStringAttr("narrow_range"), tmpAttr); } ::llvm::SmallVector<Type, 4> target_types;
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
output_types = {output_type.clone(elem_type.getStorageType())}; } SmallVector<Value> args = {q_op.getArg(), scale, zero_point}; FlatSymbolRefAttr func_name = FlatSymbolRefAttr::get(rewriter.getStringAttr(kQuantizeFuncName)); auto quantize_call = rewriter.create<TF::PartitionedCallOp>( loc, output_types, args, func_name, /*config=*/"", /*config_proto=*/"", /*executor_type=*/"");
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 54.5K bytes - Viewed (0)