Home
last modified time | relevance | path

Searched defs:cluster_func (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
Dtpu_resource_read_for_write.cc49 tf_device::ClusterFuncOp cluster_func, Value result) { in GetResourceWriteResult()
72 bool ClusterFuncHasResourceRead(tf_device::ClusterFuncOp cluster_func, in ClusterFuncHasResourceRead()
84 getOperation().walk([&](tf_device::ClusterFuncOp cluster_func) { in runOnOperation()
91 for (tf_device::ClusterFuncOp cluster_func : cluster_funcs) { in runOnOperation() local
Dtpu_resource_partitioning.cc74 void PartitionResourceReadsWrites(tf_device::ClusterFuncOp cluster_func) { in PartitionResourceReadsWrites()
144 getFunction()->walk([&cluster_funcs](tf_device::ClusterFuncOp cluster_func) { in runOnFunction()
147 for (tf_device::ClusterFuncOp cluster_func : cluster_funcs) in runOnFunction() local
Dtpu_sharding_identification_pass.cc132 tf_device::ClusterFuncOp cluster_func, FuncOp func, Builder* builder, in IdentifyXlaShardingForComputationInputs()
247 tf_device::ClusterFuncOp cluster_func, FuncOp func, Builder* builder, in IdentifyXlaShardingForComputationOutputs()
292 Builder* builder, tf_device::ClusterFuncOp cluster_func) { in IdentifyXlaShardingForTPUComputation()
371 getOperation().walk([&](tf_device::ClusterFuncOp cluster_func) { in runOnOperation()
Dmark_input_output_aliases.cc54 tf_device::ClusterFuncOp cluster_func, in BuildAliasingInfo()
127 module.walk([&](tf_device::ClusterFuncOp cluster_func) { in runOnOperation()
Dtpu_space_to_depth_pass.cc377 TF::SpaceToDepthOp BuildSpaceToDepth(tf_device::ClusterFuncOp cluster_func, in BuildSpaceToDepth()
394 tf_device::ClusterFuncOp cluster_func, in HandleHostInput()
406 tf_device::ClusterFuncOp cluster_func, in HandleHostReplicatedInputs()
434 void HandleCluster(tf_device::ClusterFuncOp cluster_func, int32_t block_size, in HandleCluster()
604 Optional<tf_device::ClusterFuncOp> cluster_func; in runOnOperation() local
Dtpu_rewrite_pass.cc330 tf_device::ClusterFuncOp cluster_func, int num_replicas, in BuildCompileOp()
436 llvm::ArrayRef<Value> inputs, tf_device::ClusterFuncOp cluster_func, in BuildExecuteOp()
457 Operation* compile_op, tf_device::ClusterFuncOp cluster_func, in BuildParallelExecuteOp()
553 tf_device::ClusterFuncOp cluster_func, in Rewrite()
Dannotate_parameter_replication.cc61 m.walk([&](tf_device::ClusterFuncOp cluster_func) { in runOnOperation()
/external/tensorflow/tensorflow/compiler/mlir/tensorflow/utils/
Dxla_sharding_util.cc210 mlir::tf_device::ClusterFuncOp cluster_func, mlir::OpBuilder* builder, in ExtractInputsForLogicalDevices()
311 mlir::tf_device::ClusterFuncOp cluster_func, in ParseAndValidateOutputSharding()
502 mlir::tf_device::ClusterFuncOp cluster_func, in GetOutputTypesForLogicalDeviceComputation()
539 mlir::tf_device::ClusterFuncOp cluster_func, in RemapOutputsFromLogicalDevices()