Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 5 of 5 for TensorListSetItemOp (0.21 sec)

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

        : public OpConversionPattern<TF::TensorListSetItemOp> {
      explicit ConvertTensorListSetItem(MLIRContext *context,
                                        bool enable_dynamic_update_slice = false)
          : OpConversionPattern<TF::TensorListSetItemOp>(context),
            enable_dynamic_update_slice(enable_dynamic_update_slice) {}
    
      LogicalResult matchAndRewrite(
          TF::TensorListSetItemOp op, OpAdaptor adaptor,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 70.7K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc

      return success();
    }
    
    LogicalResult HandleTensorListSetItemOp(
        TF::TensorListSetItemOp set_item,
        llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size) {
      auto buffer = set_item.getInputHandle();
      auto it = buffer_to_size->find(buffer);
      if (it == buffer_to_size->end()) {
        set_item.emitOpError("found tf.TensorListSetItemOp on unknown TensorList.");
        return failure();
      }
      OpBuilder builder(set_item);
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 39.2K bytes
    - Viewed (0)
  3. tensorflow/compiler/mlir/lite/transforms/legalize_tensorlist.cc

      }
      if (auto stack = llvm::dyn_cast_or_null<TF::TensorListStackOp>(op)) {
        element_type = stack.getElementDtype();
      }
      if (auto set_item = llvm::dyn_cast_or_null<TF::TensorListSetItemOp>(op)) {
        element_type = set_item.getElementDtype();
      }
      if (auto from_tensor =
              llvm::dyn_cast_or_null<TF::TensorListFromTensorOp>(op)) {
        element_type = from_tensor.getElementDtype();
      }
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue May 07 23:04:40 UTC 2024
    - 10.6K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tf2xla/internal/passes/mark_ops_for_outside_compilation.cc

          GET_OPERATION_NAME(mlir::TF::TensorListPushBackOp),
          GET_OPERATION_NAME(mlir::TF::TensorListPopBackOp),
          GET_OPERATION_NAME(mlir::TF::TensorListGetItemOp),
          GET_OPERATION_NAME(mlir::TF::TensorListSetItemOp),
          GET_OPERATION_NAME(mlir::TF::TensorListLengthOp),
          GET_OPERATION_NAME(mlir::TF::TensorListElementShapeOp),
          GET_OPERATION_NAME(mlir::TF::TensorListGatherOp),
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 21.4K bytes
    - Viewed (0)
  5. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

            if (!verify_and_update_potential_element_type(element_type))
              return false;
            add_to_worklist(scatter.getOutputHandle());
            continue;
          }
          if (auto set_item = llvm::dyn_cast<TensorListSetItemOp>(use.getOwner())) {
            auto element_type =
                mlir::dyn_cast<RankedTensorType>(set_item.getItem().getType());
            DCOMMENT("\tTensorListSetItemOp " << element_type);
    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