Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for attr_in (0.11 sec)

  1. tensorflow/c/experimental/ops/gen/cpp/views/attr_view.cc

      string c_base_type = attr_.base_type();
      if (attr_.base_type() == "type") {
        c_base_type = "DataType";
      } else if (attr_.base_type() == "shape") {
        c_base_type = "const PartialTensorShape";
      }
    
      // ... and wrap in a Span<> if it's a list.
      if (attr_.is_list()) {
        return absl::Substitute("absl::Span<$0>", c_base_type);
      } else {
        return c_base_type;
      }
    
      return attr_.full_type();
    }
    
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Jun 03 07:02:00 UTC 2024
    - 4.3K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/utils/perception_ops_utils.cc

        return func_.emitWarning()
               << "Invalid number of results from " << kMaxUnpooling << ": "
               << func_.getFunctionType().getNumResults();
      }
    
      auto attrs = attr_.getAttrs();
    
      if (failed(HasIntegerArrayWithSize(&func_, attrs, "pool_size", 2))) {
        return failure();
      }
    
      if (failed(HasIntegerArrayWithSize(&func_, attrs, "strides", 2))) {
        return failure();
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri May 17 17:58:54 UTC 2024
    - 8.9K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/utils/nms_utils.cc

                     StringAttr::get(func_.getContext(), kCustomSSDPostprocessing));
    
      OpBuilder builder(func_.getBody());
      std::string custom_option_buffer;
      if (failed(CreateNMSCustomOptions(func_, attr_.getAttrs(),
                                        custom_option_buffer))) {
        return failure();
      }
      auto op = builder.create<CustomOp>(
          func_.getLoc(), func_.getFunctionType().getResults(),
    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