Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 2 of 2 for BlockArgument (0.17 sec)

  1. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

        if (opr) {
          producer = opr.getOwner();
          port = {opr.getResultNumber()};
        } else {
          producer = mlir::cast<BlockArgument>(v);
          port = {0};
        }
      }
      ValuePort(PointerUnion<Operation*, BlockArgument> producer,
                SmallVector<unsigned int, 2> port)
          : producer(producer), port(port) {}
    
      raw_ostream& print(raw_ostream& os) const {
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/flatbuffer_export.cc

      // have associated tensor and buffer. Build FlatBuffer tensor and buffer for
      // other functions.
      for (unsigned i = 0, e = bb.getNumArguments(); i < e; ++i) {
        mlir::BlockArgument arg = bb.getArgument(i);
        std::string tensor_name;
        if (has_input_attr)
          tensor_name = std::string(name_mapper_.GetUniqueName(arg));
        if (tensor_name.empty()) tensor_name = absl::StrCat("arg", i);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Wed Jun 12 21:41:49 UTC 2024
    - 164.5K bytes
    - Viewed (0)
Back to top