Searched defs:split_shape (Results 1 – 2 of 2) sorted by relevance
173 for(const auto &split_shape : split_shapes) in compute_target() local228 for(const auto &split_shape : split_shapes) in compute_reference() local
615 auto split_shape = Reshape(scope, Transpose(scope, stack, perm), {-1}); in TileGrad() local