Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 1 of 1 for TensorListPushBackOp (0.17 sec)

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

        // (conceptually) modify it i.e., ops that take an input TensorList and
        // produce an output TensorList.
        for (auto& use : tensorlist.getUses()) {
          if (auto push = llvm::dyn_cast<TensorListPushBackOp>(use.getOwner())) {
            auto element_type =
                mlir::dyn_cast<RankedTensorType>(push.getTensor().getType());
            if (!verify_and_update_potential_element_type(element_type))
              return false;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
Back to top