Home
last modified time | relevance | path

Searched defs:cluster_ops (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtpu_cluster_formation.cc194 Operation* op, bool incoming, const OpSetVector& cluster_ops, in hasOpClusterControlDependency()
211 const OpSetVector& cluster_ops, in hasOpClusterDataDependency()
238 Block* block, const OpSetVector& cluster_ops, in CollectClusterSuccessorOps()
300 Block* block, const OpSetVector& cluster_ops) { in CollectClusterResults()
320 Block* block, const OpSetVector& cluster_ops, llvm::ArrayRef<Value> results, in CreateClusterOp()
593 const auto& cluster_ops = cluster_metadata_and_ops.getSecond(); in FormClustersInBlock() local
H A Dtpu_extract_outside_compilation.cc280 bool HasDynamicOutputs(const llvm::SmallSetVector<Operation*, 4>& cluster_ops) { in HasDynamicOutputs()
310 const llvm::SmallSetVector<Operation*, 4>& cluster_ops) { in GetStaticExternalOperands()
340 const llvm::SmallSetVector<Operation*, 4>& cluster_ops) { in GetAllExternalOperands()
359 const llvm::SmallSetVector<Operation*, 4>& cluster_ops) { in GetExternalOperands()
372 const llvm::SmallSetVector<Operation*, 4>& cluster_ops) { in GetExternalOutputs()
H A Dtpu_extract_head_tail_outside_compilation.cc141 auto cluster_ops = cluster.GetBody().without_terminator(); in FindOutsideCompiledOpsAtHead() local
243 auto cluster_ops = llvm::reverse(cluster.GetBody().without_terminator()); in FindOutsideCompiledOpsAtTailAndClusterResults() local
H A Dtpu_host_computation_expansion.cc71 auto cluster_ops = cluster.GetBody().without_terminator(); in ExpandHeadOutsideCompiledOps() local
/aosp_15_r20/system/core/fs_mgr/libsnapshot/libsnapshot_cow/
H A Dcow_format.cpp112 int64_t GetNextOpOffset(const CowOperationV2& op, uint32_t cluster_ops) { in GetNextOpOffset()
122 int64_t GetNextDataOffset(const CowOperationV2& op, uint32_t cluster_ops) { in GetNextDataOffset()
H A Dparser_v2.cpp96 uint64_t cluster_ops = header_.cluster_ops ?: 1; in ParseOps() local
H A Dtest_v3.cpp786 size_t cluster_ops; member
942 std::vector<size_t> cluster_ops = {1, 256}; in GetTestConfigs() local
H A Dtest_v2.cpp1141 int cluster_ops = 5; in TEST_F() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Ddesignate_resource_handle_mesh.cc46 const auto& cluster_ops = cluster.GetBody().without_terminator(); in SetMeshForResourceCreatingCluster() local
H A Dmesh_propagation.cc665 llvm::SmallVector<mlir::tf_device::ClusterOp, 8> cluster_ops; in PropagateMesh() local
/aosp_15_r20/system/core/fs_mgr/libsnapshot/include/libsnapshot/
H A Dcow_writer.h48 uint32_t cluster_ops = 1024; member
H A Dcow_format.h83 uint32_t cluster_ops; member