Searched defs:dims_1 (Results 1 – 6 of 6) sorted by relevance
61 int64_t dims_1[1] = {3}; in test_sum_dim_out_invalid_dimensions() local99 int64_t dims_1[1] = {1}; in test_sum_dim_out_invalid_shape() local134 int64_t dims_1[1] = {2}; in test_sum_dim_out_dtype() local288 int64_t dims_1[1] = {2}; in TEST_F() local330 int64_t dims_1[1] = {2}; in TEST_F() local
56 int64_t dims_1[1] = {3}; in test_amin_out_invalid_dimensions() local89 int64_t dims_1[1] = {1}; in test_amin_out_invalid_shape() local119 int64_t dims_1[1] = {2}; in test_amin_out_dtype() local273 int64_t dims_1[1] = {2}; in TEST_F() local
56 int64_t dims_1[1] = {3}; in test_amax_out_invalid_dimensions() local89 int64_t dims_1[1] = {1}; in test_amax_out_invalid_shape() local119 int64_t dims_1[1] = {2}; in test_amax_out_dtype() local273 int64_t dims_1[1] = {2}; in TEST_F() local
62 int64_t dims_1[1] = {3}; in test_var_out_invalid_dimensions() local100 int64_t dims_1[1] = {1}; in test_var_out_invalid_shape() local139 int64_t dims_1[1] = {2}; in test_var_out_dtype() local291 int64_t dims_1[1] = {2}; in TEST_F() local
61 int64_t dims_1[1] = {3}; in test_mean_dim_out_invalid_dimensions() local97 int64_t dims_1[1] = {1}; in test_mean_dim_out_invalid_shape() local130 int64_t dims_1[1] = {2}; in test_mean_dim_out_dtype() local304 int64_t dims_1[1] = {2}; in TEST_F() local
1440 auto dims_1 = dynamic_dimension_inference->GetDynamicSizes(operand_1, {}); in RewriteDynamicBinaryOp() local