Searched refs:dim_order (Results 1 – 2 of 2) sorted by relevance
943 std::vector<int64_t> dim_order(ndim); in Create() local944 absl::c_iota(dim_order, 0); in Create()965 absl::c_stable_sort(dim_order, in Create()969 auto inv_dim_order = InversePermutation(dim_order); in Create()974 plan->lda_.push_back(input_strides_in_bytes.at(dim_order[i])); in Create()975 plan->a_dims_.push_back(dims[dim_order[i]]); in Create()
4489 const char* dim_order = "NHWC"; // can use chars NHWC in any order. in TEST_F() member4530 for (int i = 0; i < strlen(options.dim_order); ++i) { in TEST_F()4531 char ch = options.dim_order[i]; in TEST_F()4627 options.Reset().dim_order = "NWHC"; in TEST_F()4629 options.Reset().dim_order = "WHNC"; in TEST_F()4632 options.Reset().dim_order = "HWCN"; in TEST_F()4634 options.Reset().dim_order = "CHWN"; in TEST_F()4656 options.dim_order = "WHNC"; in TEST_F()4686 options.dim_order = "CHWN"; in TEST_F()4694 options.dim_order = "HWNC"; in TEST_F()[all …]