Searched refs:dim_order (Results 1 – 2 of 2) sorted by relevance
73 std::vector<int64> dim_order(then_shape.dims()); in Compile() local74 dim_order[0] = then_shape.dims() - 1; in Compile()75 std::iota(dim_order.begin() + 1, dim_order.end(), 0); in Compile()76 cond_handle = xla::Transpose(cond_handle, dim_order); in Compile()
3289 const char* dim_order = "NHWC"; // can use chars NHWC in any order. in TEST_F() member3330 for (int i = 0; i < strlen(options.dim_order); ++i) { in TEST_F()3331 char ch = options.dim_order[i]; in TEST_F()3423 options.Reset().dim_order = "NWHC"; in TEST_F()3425 options.Reset().dim_order = "WHNC"; in TEST_F()3428 options.Reset().dim_order = "HWCN"; in TEST_F()3430 options.Reset().dim_order = "CHWN"; in TEST_F()3452 options.dim_order = "WHNC"; in TEST_F()3482 options.dim_order = "CHWN"; in TEST_F()3490 options.dim_order = "HWNC"; in TEST_F()[all …]