Searched defs:out_sizes (Results 1 – 8 of 8) sorted by relevance
33 int64* out_sizes = static_cast<int64*>(data[2]); in argmax_float_2d_xla_impl() local
95 const std::vector<int64> out_sizes = {batch, out_rows, out_cols, in Compute() local
121 const std::vector<int64> out_sizes = { in Compute() local
152 size_t out_sizes[1]; member
138 const std::vector<int64> out_sizes = {batch, out_rows, out_cols, depth}; in Compute() local
762 size_t out_sizes[4]; member
689 size_t out_sizes[4]; member
3953 int out_sizes[4]; in Transpose() local