Searched defs:keys_shape (Results 1 – 7 of 7) sorted by relevance
220 Shape keys_shape = instruction->operand(0)->shape(); in AddBackendConstraints() local
1211 Shape keys_shape = sort->operand(0)->shape(); in HandleSort() local
307 const Shape& keys_shape = values_arrays[0].GetShape(); in EmitSortInPlace() local
1071 Shape keys_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local1089 Shape keys_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
2450 Shape keys_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local2468 Shape keys_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
2804 Shape keys_shape = ShapeUtil::MakeShape(F32, {1}); in TEST_F() local2820 Shape keys_shape = ShapeUtil::MakeShape(F32, {5, 0}); in TEST_F() local
507 Shape keys_shape = sort->keys()->shape(); in HandleSort() local