Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for eraseBody (0.12 sec)

  1. tensorflow/compiler/mlir/lite/transforms/prepare_composite_functions_tf.cc

        func.eraseBody();
        func.addEntryBlock();
        convert_embedded_lookup.RewriteFunc();
      } else if (attr.getValue() == mlir::TFL::kLstmCellSimple) {
        // Check if the lstm cell simple can be fused, if not, we just don't do
        // anything.
        if (failed(CheckFusableLstmCellSimple(func))) return;
        func.eraseBody();
        func.addEntryBlock();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 17.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/utils/tftext_utils.cc

        }
      }
    
      return success();
    }
    
    LogicalResult ConvertWhitespaceTokenizer(func::FuncOp func, llvm::StringRef api,
                                             FuncAttr attr) {
      func.eraseBody();
      func.addEntryBlock();
      func->setAttr(kTFImplements, attr);
      OpBuilder builder(func.getBody());
      std::string empty_option_buffer;
      auto op = builder.create<CustomOp>(
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 14.4K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/utils/perception_ops_utils.cc

                 << " does not contain integer values";
        }
        results->push_back(value.getInt());
      }
      return success();
    }
    
    }  // namespace
    
    LogicalResult ConvertMaxUnpoolingFunc::RewriteFunc() {
      func_.eraseBody();
      func_.addEntryBlock();
      func_->setAttr(kTFImplements,
                     StringAttr::get(func_.getContext(), kMaxUnpooling));
    
      OpBuilder builder(func_.getBody());
      std::string custom_option_buffer;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 8.9K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/lite/transforms/legalize_jax_random.cc

      if (!result_tuple_ty) return;
      if (result_tuple_ty.size() != 1) return;
      auto result_ty = mlir::dyn_cast<ShapedType>(result_tuple_ty.getType(0));
    
      func.eraseBody();
      func.addEntryBlock();
      ImplicitLocOpBuilder builder(func.getLoc(), func.getBody());
      llvm::SmallVector<int32_t> result_shape_i32;
      auto result_shape = result_ty.getShape();
      for (auto element : result_shape) {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 5.1K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/lite/utils/nms_utils.cc

                                      "non_max_suppression_padded";
      }
      return success();
    }
    
    LogicalResult ConvertSSDPostProcessFunc::RewriteFunc() {
      func_.eraseBody();
      func_.addEntryBlock();
      func_->setAttr(kTFImplements,
                     StringAttr::get(func_.getContext(), kCustomSSDPostprocessing));
    
      OpBuilder builder(func_.getBody());
      std::string custom_option_buffer;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 8.1K bytes
    - Viewed (0)
Back to top