Searched defs:keys_shape (Results 1 – 6 of 6) sorted by relevance
272 Shape keys_shape = instruction->operand(0)->shape(); in AddBackendConstraints() local
2707 const Shape& keys_shape = context.operand_shapes[0]; in EmitSort() local
309 const Shape& keys_shape = values_arrays[0].GetShape(); in EmitSortInPlace() local
2645 Shape keys_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local2663 Shape keys_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
3597 Shape keys_shape = ShapeUtil::MakeShape(F32, {1}); in TEST_F() local3613 Shape keys_shape = ShapeUtil::MakeShape(F32, {5, 0}); in TEST_F() local
548 Shape keys_shape = sort->keys()->shape(); in HandleSort() local