- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 35 for getNamedAttr (0.5 sec)
-
tensorflow/compiler/mlir/mlir_graph_optimization_pass_test.cc
const FunctionLibraryDefinition& function_library) override { mlir::Builder b(module.getContext()); auto producer = b.getNamedAttr("producer", b.getI32IntegerAttr(0)); auto min_consumer = b.getNamedAttr("min_consumer", b.getI32IntegerAttr(0)); auto bad_consumers = b.getNamedAttr("bad_consumers", b.getI32ArrayAttr({1, 2, 3, 4})); module->setAttr("tf.versions",
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/lite/stablehlo/transforms/smuggle_disallowed_ops.cc
// align_corners = false, device = "", half_pixel_centers = true} LogicalResult SmuggleOp(Operation* op, PatternRewriter& rewriter) { auto call_target = rewriter.getNamedAttr("call_target_name", op->getName().getIdentifier()); SmallVector<NamedAttribute> attrs{op->getAttrs()}; attrs.push_back(call_target); auto custom_call = rewriter.create<mlir::stablehlo::CustomCallOp>(
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 19 01:14:04 UTC 2024 - 3.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tpu_device_propagation.cc
if (attr.getName() == kFuncDeviceAttr) { resultAttrForOp[i] = builder.getNamedAttr( kFuncDeviceAttr, StringAttr::get(func.getContext(), it->getSecond())); found = true; break; } } if (!found) { resultAttrForOp.push_back(builder.getNamedAttr( kFuncDeviceAttr,
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/tensorflow/transforms/tpu_dynamic_layout_pass.cc
llvm::ArrayRef<Value>{compilation_key}, llvm::ArrayRef<NamedAttribute>{ builder->getNamedAttr("index", builder->getI64IntegerAttr(execute_arg_index)), builder->getNamedAttr("is_output", builder->getBoolAttr(false))}); } // Builds a TPUCopyWithLayoutOp with the given get_layout op and input.
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/tensorflow/transforms/tfg-to-tfe.cc
if (!graph) continue; Builder b(&context); auto producer = b.getNamedAttr( "producer", b.getI32IntegerAttr(graph.getVersion().getProducer())); auto min_consumer = b.getNamedAttr( "min_consumer", b.getI32IntegerAttr(graph.getVersion().getMinConsumer())); auto bad_consumers = b.getNamedAttr("bad_consumers",
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 21.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/utils/convert_attr.cc
for (const auto& func_attr : value.func().attr()) { TF_ASSIGN_OR_RETURN(auto attr, ConvertAttributeValue(func_attr.second, builder)); attrs.push_back(builder->getNamedAttr(func_attr.first, attr)); } auto func_attrs = builder->getDictionaryAttr(attrs); return mlir::TF::FuncAttr::get(builder->getContext(), value.func().name(), func_attrs);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Apr 26 09:37:10 UTC 2024 - 4.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/flatbuffer_import.cc
return llvm::SmallVector<mlir::NamedAttribute, 4>{ builder.getNamedAttr("then_branch", then_attr), builder.getNamedAttr("else_branch", else_attr), // TODO(b/139667752): Analyze statelessness correctly builder.getNamedAttr("is_stateless", builder.getBoolAttr(false))}; } if (auto* opts = options.AsWhileOptions()) {
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/legalize_tf.cc
attributes.push_back(rewriter.getNamedAttr("fused_activation_function", rewriter.getStringAttr("TANH"))); // cell_clip. attributes.push_back( rewriter.getNamedAttr("cell_clip", rewriter.getF32FloatAttr(0.0))); // proj_clip. attributes.push_back( rewriter.getNamedAttr("proj_clip", rewriter.getF32FloatAttr(0.0)));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Mon May 20 20:06:54 UTC 2024 - 45.2K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc
llvm::ArrayRef<Value>{}, llvm::ArrayRef<NamedAttribute>{ builder->getNamedAttr(kDeviceAttr, builder->getStringAttr(device)), builder->getNamedAttr("container", builder->getStringAttr("")), builder->getNamedAttr( "shared_name", builder->getStringAttr(GetRandomStateVariableName()))})); } return state_vars; }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/translate/import_model.cc
auto inputs = b.getNamedAttr("inputs", b.getStringAttr(ss.str())); s.clear(); llvm::interleave(ret_nodes, ss, node_name, ","); auto outputs = b.getNamedAttr("outputs", b.getStringAttr(ss.str())); s.clear(); llvm::interleave(specs.control_outputs, ss, ","); auto control_outputs = b.getNamedAttr("control_outputs", b.getStringAttr(ss.str()));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed May 01 11:17:36 UTC 2024 - 183.2K bytes - Viewed (0)