Search Options

Results per page
Sort
Preferred Languages
Advance

Results 1 - 7 of 7 for TensorListFromTensorOp (0.26 sec)

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

      (*buffer_to_size)[buffer] = {size, /*fixed=*/true};
      list.getHandle().replaceAllUsesWith(buffer);
      list.erase();
      return success();
    }
    
    LogicalResult HandleTensorListFromTensorOp(
        TF::TensorListFromTensorOp list,
        llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size) {
      OpBuilder builder(list);
      Value buffer = builder.create<TF::IdentityOp>(
          list.getLoc(), ArrayRef<Type>{list.getTensor().getType()},
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu Apr 25 16:01:03 UTC 2024
    - 39.2K bytes
    - Viewed (0)
  2. tensorflow/compiler/mlir/lite/transforms/legalize_tensorlist.cc

      }
      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();
      }
      if (auto get_item = llvm::dyn_cast_or_null<TF::TensorListGetItemOp>(op)) {
        element_type = get_item.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)
  3. tensorflow/compiler/mlir/quantization/tensorflow/passes/duplicate_shape_determining_constants.cc

          CompileTimeConstantOperand<TF::TensorArraySplitV3Op, 2>,  // $lengths
          CompileTimeConstantOperand<TF::TensorArrayV3Op, 0>,       // $size
          // $element_shape
          CompileTimeConstantOperand<TF::TensorListFromTensorOp, 1>,
          // $element_shape, $num_elements
          CompileTimeConstantOperand<TF::TensorListReserveOp, 0, 1>,
          // $begin, $end, $strides
          CompileTimeConstantOperand<TF::TensorStridedSliceUpdateOp, 1, 2, 3>,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Fri Mar 22 05:52:39 UTC 2024
    - 17.5K bytes
    - Viewed (0)
  4. tensorflow/compiler/mlir/tf2xla/internal/passes/mark_ops_for_outside_compilation.cc

          // 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),
          GET_OPERATION_NAME(mlir::TF::TensorListGetItemOp),
    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/ir/tf_ops_n_z.cc

    // TensorListGetItemOp
    //===----------------------------------------------------------------------===//
    
    namespace {
    // If the input of TensorListGetItemOp is TensorListFromTensorOp and the
    // TensorListFromTensorOp is only used by TensorListGetItemOp (not modified by
    // other TensorList ops), we can convert it to a GatherOp.
    class ConvertTensorListGetItemOpOfTensorListFromTensorOpToGather
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Thu May 09 22:07:10 UTC 2024
    - 170.8K bytes
    - Viewed (0)
  6. tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc

    }
    
    // Returns true if the op creates a TensorList.
    bool IsTensorListInitOp(Operation* op) {
      return isa<TensorListReserveOp>(op) || isa<EmptyTensorListOp>(op) ||
             isa<TensorListFromTensorOp>(op);
    }
    
    // Returns the `element_shape` operand of the ops that create a TensorList.
    Value GetElementShapeOperand(Operation* op) {
      if (auto empty_tl = dyn_cast<EmptyTensorListOp>(op))
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Sat Jun 08 07:28:49 UTC 2024
    - 134.1K bytes
    - Viewed (0)
  7. tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc

                   .getValue();
      };
    
      ConversionTarget target(*context);
      target.addDynamicallyLegalDialect<TF::TensorFlowDialect>(is_legal);
      target.addIllegalOp<TF::EmptyTensorListOp, TF::TensorListFromTensorOp,
                          TF::TensorListGetItemOp, TF::TensorListLengthOp,
                          TF::TensorListPushBackOp, TF::TensorListReserveOp,
                          TF::TensorListStackOp, TF::TensorListResizeOp,
    Registered: Sun Jun 16 05:45:23 UTC 2024
    - Last Modified: Tue Jun 11 20:00:43 UTC 2024
    - 70.7K bytes
    - Viewed (0)
Back to top