Searched defs:keys_shape (Results 1 – 6 of 6) sorted by relevance
253 Shape keys_shape = instruction->operand(0)->shape(); in AddBackendConstraints() local
2862 const Shape& keys_shape = context.operand_shapes[0]; in EmitSortFromMlir() local
309 const Shape& keys_shape = values_arrays[0].GetShape(); in EmitSortInPlace() local
2567 Shape keys_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local2585 Shape keys_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
3564 Shape keys_shape = ShapeUtil::MakeShape(F32, {1}); in TEST_F() local3580 Shape keys_shape = ShapeUtil::MakeShape(F32, {5, 0}); in TEST_F() local
548 Shape keys_shape = sort->keys()->shape(); in HandleSort() local