Searched defs:keys_shape (Results 1 – 6 of 6) sorted by relevance
247 Shape keys_shape = instruction->operand(0)->shape(); in AddBackendConstraints() local
1032 Shape keys_shape = sort->operand(0)->shape(); in HandleSort() local
309 const Shape& keys_shape = values_arrays[0].GetShape(); in EmitSortInPlace() local
2544 Shape keys_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local2562 Shape keys_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
3190 Shape keys_shape = ShapeUtil::MakeShape(F32, {1}); in TEST_F() local3206 Shape keys_shape = ShapeUtil::MakeShape(F32, {5, 0}); in TEST_F() local
566 Shape keys_shape = sort->keys()->shape(); in HandleSort() local