- Sort Score
- Result 10 results
- Languages All
Results 1 - 5 of 5 for TensorListSetItemOp (0.21 sec)
-
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) -
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) -
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) -
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) -
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)