Searched defs:keys_shape (Results 1 – 6 of 6) sorted by relevance
338 Shape keys_shape = instruction->operand(0)->shape(); in AddBackendConstraints() local
2562 const Shape& keys_shape = GetShape(operands[0]); in EmitSort() local
333 const Shape& keys_shape = values_arrays[0].GetShape(); in EmitSortInPlace() local
2628 Shape keys_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local2646 Shape keys_shape = ShapeUtil::MakeShape(F32, {8}); in TEST_F() local
4000 Shape keys_shape = ShapeUtil::MakeShape(F32, {1}); in TEST_F() local4016 Shape keys_shape = ShapeUtil::MakeShape(F32, {5, 0}); in TEST_F() local
562 Shape keys_shape = sort->keys()->shape(); in HandleSort() local