- Sort Score
- Result 10 results
- Languages All
Results 21 - 30 of 31 for Dadd (0.03 sec)
-
tensorflow/c/experimental/stream_executor/stream_executor.cc
if (memory_bandwidth >= 0) { builder.set_memory_bandwidth(memory_bandwidth); } } // TODO(annarev): Add gflops field in DeviceDescription and set it here. // TODO(annarev): Perhaps add `supports_unified_memory` in // DeviceDescription. return builder.Build(); } absl::StatusOr<std::unique_ptr<Event>> CreateEvent() override {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri Jun 14 07:39:19 UTC 2024 - 27.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tfr/passes/raise_to_tf.cc
OperationState new_state(loc, tf_op_name, inputs, output_types, attr_list); Operation* new_op = rewriter.create(new_state); if (materialize_derived_attrs_) { for (const auto& attr : derived_attrs) { // Add or update the derived attribute with the value. Skip the fixed // element type attributes, in case they are present in the NodeDef. if (!fixed_elt_type_attrs_.contains(attr.first())) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 21.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/stablehlo/passes/defer_activation_transpose.cc
}; void DeferActivationTransposePass::runOnOperation() { func::FuncOp func_op = getOperation(); MLIRContext& ctx = getContext(); RewritePatternSet patterns(&ctx); patterns.add<DeferActivationTransposeForAddOp, DeferActivationTransposeForMaxPoolReduceWindowOp, DeferActivationTransposeForMaxOp>(&ctx); if (failed(applyPatternsAndFoldGreedily(func_op, std::move(patterns)))) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/prepare_quantize_drq.cc
void PrepareQuantizeDRQPass::runOnOperation() { MLIRContext* ctx = &getContext(); RewritePatternSet patterns(ctx); ModuleOp module_op = getOperation(); populateWithGenerated(patterns); patterns.add<PrepareDRQQuantizableOp>(ctx, quant_specs_, op_set_, enable_per_channel_quantization_); FrozenRewritePatternSet frozen_patterns(std::move(patterns));
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 11.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/sparsecore/embedding_sequencing.cc
OpType& found_op) { // Find the TPUReplicationMetadata or TPUCompilationResult ops which will be // cloned/inserted into each region. We add them to the merged_set so that // they're ignored when extracting the four main functions. found_op = nullptr; for (OpType op : func.getOps<OpType>()) { if (found_op != nullptr) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 39.4K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/decompose_reduce_dataset.cc
// Then branch gets the data and calls the reduce_function. auto& then_branch = dataset_if.getThenBranch(); then_branch.push_back(new Block); builder.setInsertionPointToEnd(&then_branch.front()); // Add iterator operational data access inside if. auto get_value = builder.create<TF::OptionalGetValueOp>(loc, dataset_types, get_next.getResult());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/fused_kernel_matcher.cc
}); return false; } return true; } }; void FusedKernelMatcherPass::runOnOperation() { RewritePatternSet patterns(&getContext()); auto func = getOperation(); patterns.add<FuseConv2DBiasAdd, FuseMatMulBiasAdd>(&getContext()); (void)applyPatternsAndFoldGreedily(func, std::move(patterns)); } } // namespace
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 14.9K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/internal/passes/extract_head_tail_outside_compilation.cc
// Insert all ops including nested ops for checking outputs/side effects. cluster_op.walk( [&](Operation* op) { tail_outside_compiled_ops_set.insert(op); }); // Only add top level ops to output vector. tail_outside_compiled_ops->push_back(&cluster_op); } *cluster_results = cluster_results_set.takeVector(); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 20.6K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_saved_model.cc
return success(); } }; void SessionInitializerOp::getCanonicalizationPatterns( RewritePatternSet &results, MLIRContext *context) { results.add<OptimizeSessionInitializerPattern>(context); } SmallVector<StringRef, 2> GetSessionInitializerExportedName(ModuleOp op) { auto session_initializer_op = GetSessionInitializerOp(op);
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 23.5K bytes - Viewed (0) -
tensorflow/compiler/mlir/tf2xla/transforms/legalization_op_config.cc
TypeID::get<TF::XlaSpmdFullToShardShapeOp>(), TypeID::get<TF::XlaSpmdShardToFullShapeOp>(), TypeID::get<TF::XlaSvdOp>(), }; // Add the ops from the TPUEmbeddingOpsRegistry. for (auto op_type_id : TF::TPUEmbeddingOpsRegistry::Global().GetOpsTypeIds()) { ops_set->insert(op_type_id); } return ops_set; }();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Apr 24 04:08:35 UTC 2024 - 21.7K bytes - Viewed (0)