Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for insertion_guard (0.24 sec)

  1. 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)
  2. 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)
Back to top