- Sort Score
- Result 10 results
- Languages All
Results 31 - 37 of 37 for Dadd (0.07 sec)
-
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/jit/encapsulate_xla_computations_pass.cc
AddNodeAttr("function", function, &def); for (Node* node : nodes_to_remove) { VLOG(2) << "Deleting node " << node->DebugString(); // Ensure that we do not attempt to add control edges to nodes that are // deleted. control_inputs.erase(node); control_outputs.erase(node); graph->RemoveNode(node); }
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Mar 12 06:33:33 UTC 2024 - 15.1K 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) -
tensorflow/compiler/aot/codegen.cc
std::back_inserter(result), [](const BufferInfo& buffer_info) { return buffer_info.is_temp_buffer(); }); return result; } // Add (from,to) rewrite pairs based on the given shape. These rewrite pairs // are used to generate methods for args and results. Status AddRewritesForShape(int i, const xla::Shape& shape,
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 02 01:20:01 UTC 2024 - 36.8K bytes - Viewed (0)