Home
last modified time | relevance | path

Searched defs:function_block (Results 1 – 2 of 2) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dtpu_sharding_identification_pass.cc114 Block& function_block = func.front(); in VerifyShardings() local
187 Block& function_block = func.front(); in IdentifyXlaShardingForComputationInputs() local
357 Block& function_block = func.front(); in IdentifyXlaShardingForComputationOutputs() local
471 Block& function_block = func.front(); in IdentifyXlaShardingForTPUComputation() local
/aosp_15_r20/external/tensorflow/tensorflow/dtensor/mlir/
H A Dtpu_integration.cc106 auto& function_block = function->getCallableRegion()->front(); in CreateTPUCluster() local