tensorflow/compiler/mlir/tensorflow/transforms/tpu_space_to_depth_pass.cc
// Performs transformation for a non-replicated input.
Performs transformation for a non-replicated input.
Performs transformation for a non - replicated input .
TF::SpaceToDepthOp HandleHostInput(Value input, int64_t index,
                                   tf_device::ClusterFuncOp cluster_func,
                                   int32_t block_size,
                                   ArrayRef<int64_t> input_shape) {
  auto space_to_depth =
      BuildSpaceToDepth(cluster_func, input, block_size, input_shape);
  cluster_func.setOperand(index, space_to_depth);
  return space_to_depth;
}
TF :: SpaceToDepthOp HandleHostInput ( Value input , int64_t index , tf_device :: ClusterFuncOp cluster_func , int32_t block_size , ArrayRef < int64_t > input_shape ) { auto space_to_depth = BuildSpaceToDepth ( cluster_func , input , block_size , input_shape ) ; cluster_func . setOperand ( index , space_to_depth ) ; return space_to_depth ; }
Performs transformation for a non-replicated input.
Performs transformation for a non - replicated input .