Searched refs:compute_shape_embedding (Results 1 – 2 of 2) sorted by relevance
38 TORCH_API std::vector<torch::lazy::Shape> compute_shape_embedding(const at::Tensor & weight, const …
380 std::vector<Shape> compute_shape_embedding( in compute_shape_embedding() function