Home
last modified time | relevance | path

Searched refs:PartitionedCallTensorArrayOpsInfo (Results 1 – 1 of 1) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtensor_array_ops_decomposition.cc514 struct PartitionedCallTensorArrayOpsInfo { struct
553 llvm::StringMap<PartitionedCallTensorArrayOpsInfo>*);
557 llvm::StringMap<PartitionedCallTensorArrayOpsInfo>* in HandleWhileOp()
642 llvm::StringMap<PartitionedCallTensorArrayOpsInfo>* in HandleIfOp()
725 llvm::StringMap<PartitionedCallTensorArrayOpsInfo>* in HandlePartitionedCallOp()
728 callee.getName(), PartitionedCallTensorArrayOpsInfo()); in HandlePartitionedCallOp()
835 llvm::StringMap<PartitionedCallTensorArrayOpsInfo>* in HandleRegionControlFlowOps()
867 llvm::StringMap<PartitionedCallTensorArrayOpsInfo>* in DecomposeTensorArrayOps()
942 llvm::StringMap<PartitionedCallTensorArrayOpsInfo> in runOnOperation()