Searched refs:size_vec (Results 1 – 1 of 1) sorted by relevance
405 auto size_vec = size->vec<int32>(); in __anonc970e8b60602() local407 c->ReplaceDim(input, 1, c->MakeDim(size_vec(0)), &input)); in __anonc970e8b60602()409 c->ReplaceDim(input, 2, c->MakeDim(size_vec(1)), &input)); in __anonc970e8b60602()440 auto size_vec = size->vec<int64>(); in __anonc970e8b60702() local441 h = c->MakeDim(size_vec(0)); in __anonc970e8b60702()442 w = c->MakeDim(size_vec(1)); in __anonc970e8b60702()