Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 4 of 4 for TensorListReserveOp (0.26 sec)

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

    // on `tf.TensorList*` ops.
    std::optional<ConstBytesAttr> CustomOptions(MLIRContext* context,
                                                mlir::Operation* op) {
      if (auto reserve =
              llvm::dyn_cast_or_null<mlir::TF::TensorListReserveOp>(op)) {
        tflite::TensorType tflite_type =
            tflite::ConvertTypeToTensorType(reserve.getElementDtype());
    
        return CreateListReserveOptions(context, tflite_type);
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 23:04:40 UTC 2024
    - 10.6K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/quantization/tensorflow/passes/duplicate_shape_determining_constants.cc

          // $element_shape
          CompileTimeConstantOperand<TF::TensorListFromTensorOp, 1>,
          // $element_shape, $num_elements
          CompileTimeConstantOperand<TF::TensorListReserveOp, 0, 1>,
          // $begin, $end, $strides
          CompileTimeConstantOperand<TF::TensorStridedSliceUpdateOp, 1, 2, 3>,
          CompileTimeConstantOperand<TF::TileOp, 1>,                // $multiples
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 22 05:52:39 UTC 2024
    - 17.5K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc

      list.getHandle().replaceAllUsesWith(buffer);
      (*buffer_to_size)[buffer] = {size, /*fixed=*/false};
      list.erase();
      return success();
    }
    
    LogicalResult HandleTensorListReserveOp(
        TF::TensorListReserveOp list,
        llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size) {
      Value buffer;
      OpBuilder builder(list);
      llvm::SmallVector<int64_t, 8> element_shape;
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 39.2K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tf2xla/internal/passes/mark_ops_for_outside_compilation.cc

          GET_OPERATION_NAME(mlir::TF::TensorArrayScatterV3Op),
          // Tensor List Ops.
          GET_OPERATION_NAME(mlir::TF::EmptyTensorListOp),
          GET_OPERATION_NAME(mlir::TF::TensorListReserveOp),
          GET_OPERATION_NAME(mlir::TF::TensorListFromTensorOp),
          GET_OPERATION_NAME(mlir::TF::TensorListPushBackOp),
          GET_OPERATION_NAME(mlir::TF::TensorListPopBackOp),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 21.4K bytes
    - Viewed (0)
Back to top