- Sort Score
- Result 10 results
- Languages All
Results 11 - 20 of 37 for Dadd (0.03 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/prepare_tpu_computation_for_tf_export.cc
// If the parent is not a FuncOp, then add the parent op containing a region // to worklist. Operation* parent = region->getParentOp(); if (!isa<func::FuncOp>(parent)) { if (ops_with_tokens.insert(parent).second) { worklist.push_back(parent); } continue; } // For functions, get all the users and add them to the worklist.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/promote_resources_to_args.cc
function.setType(builder.getFunctionType(argument_types, result_types)); // Add resource_name attribute to the output for the resources. for (auto& resource : output_only_resources) function.setResultAttr(resource.first, kResourceNameArgAttr, builder.getStringAttr(resource.second)); // Add aliasing_output attribute to the input argument for the resources that // are updated by the function.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 17.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_lifting.cc
MLIRContext* ctx = &getContext(); auto func = getOperation(); // The pattern includes decomposing batch normalization ops, fusing add/mul // with a constant operand to a preceding affine operation. RewritePatternSet patterns(ctx); populateWithGenerated(patterns); patterns.add<RemoveIdentity, ConstantFoldQuantizableOperands>(ctx); if (op_set_ != OpSet::XLA) { // Convert Einsum into BatchMatMul for non-XLA opsets.
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 13.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/lite/transforms/legalize_tf.cc
}; void addPatterns(MLIRContext* context, RewritePatternSet& patterns, bool preserve_assert_op) { // Add TF->TF lowering patterns. TF::PopulateLoweringTFPatterns(context, &patterns); // Add the generated patterns to the list. populateWithGenerated(patterns); patterns.add<ConvertTFConcatV2Op, ConvertTFBatchMatMulOp, ConvertTFBatchMatMulV2Op, ConvertTFBatchMatMulV3Op,
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/quantization/tensorflow/passes/insert_custom_aggregation_ops.cc
std::optional<StringRef> composite_function_name = GetCompsiteFunctionName(defining_op); if (!composite_function_name.has_value()) continue; // Do not add CustomAggregatorOp after Gather since it is a weight-only // quantizable op. if (auto call_op = dyn_cast_or_null<TF::PartitionedCallOp>(defining_op)) { StringRef function_name =
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/ir/tf_executor.cc
return success(); } }; // TODO(lyandy): Add canonicalization for empty IslandOps with more than one // control operand and no data results. } // anonymous namespace void IslandOp::getCanonicalizationPatterns(RewritePatternSet &results, MLIRContext *context) { results.add<DropEmptyIslandNoOperandNoDataResult,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 42.7K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/cluster_ops_by_policy.cc
// operation dominates all the users (%3 in this example). Operation *insertion_point; // After construction: // A set of constraints on the clustered operation operands that must be // satisfied in order to add operation to the cluster. For basic block // source this will be always empty. // // During the union-find cluster formation: // The root member will have constraints merged from all of the cluster
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 27.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_variable_runtime_reformatting.cc
devices.insert({device_alias, device_list_for_alias}); } OpBuilder builder(replicate); builder.setInsertionPoint(while_op); // Create per-device variables for formatting state, and add them to the while // loop. auto key_type = RankedTensorType::get({2}, TF::StringType::get(builder.getContext())); auto state_vars = CreateStateVars(devices, while_op.getLoc(), key_type, &builder);
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/jit/encapsulate_util.cc
edges.push_back(EdgeInfo{e->dst_input(), e->dst()->id()}); VLOG(4) << "Oc -> oc edge: " << e->DebugString(); } } // Remove the edge from host to outside compilation. Add a placeholder as // outside compilation node input. std::map<std::pair<string, int>, Node*> placeholders; for (int i = 0, end = edges.size(); i < end; i++) { Node* dst = g->FindNodeId(edges[i].dst_node_id);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 15.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/host_runtime/tpu_rewrite_pass.cc
auto compilation_status_type = RankedTensorType::get({}, builder->getType<TF::StringType>()); auto program_type = RankedTensorType::get({3}, builder->getType<TF::StringType>()); // Add MLIR module's fingerprint to compile metadata. uint64_t mlir_fingerprint = tensorflow::Fingerprint64(txt_module); metadata.set_mlir_fingerprint(mlir_fingerprint); std::string txt_metadata;
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Apr 30 21:25:12 UTC 2024 - 29.7K bytes - Viewed (0)