Searched refs:BuildSpaceToDepth (Results 1 – 1 of 1) sorted by relevance
423 TF::SpaceToDepthOp BuildSpaceToDepth(tf_device::ClusterFuncOp cluster_func, in BuildSpaceToDepth() function444 BuildSpaceToDepth(cluster_func, input, block_size, input_shape); in HandleHostInput()471 BuildSpaceToDepth(cluster_func, input, block_size, input_shape); in HandleHostReplicatedInputs()