Searched refs:normalized_tensor_size (Results 1 – 1 of 1) sorted by relevance
180 std::array<int32_t, 3> normalized_tensor_size; in get_normalized_tensor_size() local181 normalized_tensor_size[0] = 1; in get_normalized_tensor_size()182 normalized_tensor_size[1] = a.size(broadcast_dim); in get_normalized_tensor_size()183 normalized_tensor_size[2] = 1; in get_normalized_tensor_size()185 normalized_tensor_size[0] *= a.size(i); in get_normalized_tensor_size()188 normalized_tensor_size[2] *= a.size(i); in get_normalized_tensor_size()190 return normalized_tensor_size; in get_normalized_tensor_size()