- Sort Score
- Result 10 results
- Languages All
Results 1 - 7 of 7 for TensorListReserveOp (0.47 sec)
-
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) -
tensorflow/compiler/mlir/lite/transforms/lower_static_tensor_list.cc
rewriter.replaceOpWithNewOp<TF::FillOp>(op, result_type, list_shape, zero); return success(); } }; struct ConvertTensorListReserve : public ConvertTensorListInitOp<TF::TensorListReserveOp> { explicit ConvertTensorListReserve(MLIRContext *context, bool allow_tensorlist_pass_through, bool default_to_single_batch)
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Tue Jun 11 20:00:43 UTC 2024 - 70.7K bytes - Viewed (0) -
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) -
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) -
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) -
tensorflow/compiler/mlir/tensorflow/transforms/shape_inference.cc
return UnrankedTensorType::get(element_type); } // 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) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Sat Jun 08 07:28:49 UTC 2024 - 134.1K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/ir/tf_ops_n_z.cc
context); } //===----------------------------------------------------------------------===// // TensorListReserveOp //===----------------------------------------------------------------------===// LogicalResult TensorListReserveOp::verify() { TensorListReserveOp op = *this; // This is required to populate derived attributes during export in a
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu May 09 22:07:10 UTC 2024 - 170.8K bytes - Viewed (0)