- Sort Score
- Result 10 results
- Languages All
Results 1 - 2 of 2 for insertion_guard (0.24 sec)
-
tensorflow/compiler/mlir/tf2xla/transforms/utils.h
namespace mhlo { // Builds body for reduce op by using the template binary op as the // reducer op. template <typename Op> void BuildReduceBody(Type element_type, Region* body, OpBuilder* builder) { OpBuilder::InsertionGuard guard(*builder); Block* block = builder->createBlock(body); // Block arguments are scalars of the given element type. Type type = RankedTensorType::get(/*shape=*/{}, element_type); Location loc = body->getLoc();
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Wed Sep 06 19:12:29 UTC 2023 - 2.3K bytes - Viewed (0) -
tensorflow/compiler/mlir/quantization/tensorflow/passes/lift_hashtable_ops_as_args.cc
block.getArgumentTypes(), func_type.getResults())); IRMapping mapping; OpBuilder builder(module_op); OpBuilder::InsertionGuard g(builder); // The function has been checked to have at least one use. auto function_uses = SymbolTable::getSymbolUses(target_func, &module_op.getBodyRegion());
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Fri May 17 17:58:54 UTC 2024 - 8.2K bytes - Viewed (0)