Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 3 of 3 for TensorListPopBack (0.81 sec)

  1. tensorflow/compiler/mlir/tensorflow/tests/tensor_list_ops_decomposition.mlir

      // CHECK: "tf.Slice"
      %pop:2 = "tf.TensorListPopBack"(%1#0, %elem_shape) : (tensor<!tf_type.variant<tensor<f32>>>, tensor<0xi32>) -> (tensor<!tf_type.variant<tensor<f32>>>, tensor<f32>)
      // CHECK-NOT: tf.TensorListPopBack
      // CHECK: return
      func.return
    }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Mon Oct 30 06:52:55 UTC 2023
    - 38.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/transforms/legalize_tensorlist.cc

        // op into an op with a single varidic output in `.td`.
        auto converted = rewriter.create<TFL::CustomOp>(
            op->getLoc(), op->getResultTypes(), op->getOperands(),
            "TensorListPopBack", TFL::ConstBytesAttr::get(getContext(), ""));
        rewriter.replaceOp(op, converted.getResults());
        return success();
      }
    };
    
    struct ConvertTensorListPushBack
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 23:04:40 UTC 2024
    - 10.6K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc

        llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size) {
      auto buffer = pop.getInputHandle();
      auto it = buffer_to_size->find(buffer);
      if (it == buffer_to_size->end()) {
        pop.emitOpError("found tf.TensorListPopBack on unknown TensorList.");
        return failure();
      }
      if (it->getSecond().fixed) {
        return pop.emitError("cannot pop on a fixed-size tensor list");
      }
      auto size = it->getSecond().size;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 39.2K bytes
    - Viewed (0)
Back to top