- Sort Score
- Result 10 results
- Languages All
Results 1 - 3 of 3 for HandleWhileOp (0.11 sec)
-
tensorflow/compiler/mlir/tensorflow/transforms/stack_ops_decomposition.cc
llvm::StringMap<PartitionedCallStackOpsInfo>*); // Handles stack usage by a tf.While. It will convert the body and conditional // function signatures, and performs stack ops decomposition on them. LogicalResult HandleWhileOp( TF::WhileOp while_op, ModuleOp module, const llvm::SmallDenseMap<Value, Value>& data_var_to_size_var, llvm::StringMap<PartitionedCallStackOpsInfo>* decomposed_partitioned_call_callees) {
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Apr 25 16:01:03 UTC 2024 - 23.8K bytes - Viewed (0) -
tensorflow/compiler/mlir/tensorflow/transforms/tensor_list_ops_decomposition.cc
auto output_buffer_to_size = AddTensorListSizesToTerminator<func::ReturnOp>( func.front(), buffer_to_size); UpdateFuncType(func); return output_buffer_to_size; } LogicalResult HandleWhileOp( TF::WhileOp while_op, ModuleOp module, llvm::SmallDenseMap<Value, SizeInfo>* buffer_to_size, llvm::StringMap<PartitionedCallDecompositionInfo>* decomposed_partitioned_call_callees) {
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/tensorflow/transforms/tensor_array_ops_decomposition.cc
UpdateFuncType(func); } LogicalResult DecomposeTensorArrayOps( Block*, ModuleOp, llvm::SmallDenseMap<Value, TensorArrayStats>*, llvm::StringMap<PartitionedCallTensorArrayOpsInfo>*); LogicalResult HandleWhileOp(TF::WhileOp while_op, ModuleOp module, llvm::SmallDenseMap<Value, TensorArrayStats>* stats, llvm::StringMap<PartitionedCallTensorArrayOpsInfo>*
Registered: Sun Jun 16 05:45:23 UTC 2024 - Last Modified: Thu Mar 02 20:41:19 UTC 2023 - 40.2K bytes - Viewed (0)