HomeSort by relevance Sort by last modified time
    Searched defs:new_shapes (Results 1 - 2 of 2) sorted by null

  /external/tensorflow/tensorflow/compiler/xla/service/
sort_simplifier.cc 70 std::vector<Shape> new_shapes; local
74 new_shapes.push_back(sort->operand(i)->shape());
78 Shape new_sort_shape = new_shapes.size() == 1
79 ? new_shapes[0]
80 : ShapeUtil::MakeTupleShape(new_shapes);
119 new_shapes[new_index], new_sort, new_index));
stable_sort_expander.cc 101 std::vector<Shape> new_shapes = sort->operand_count() == 1 local
104 new_shapes.push_back(iota_shape);
105 Shape new_sort_shape = ShapeUtil::MakeTupleShape(new_shapes);

Completed in 268 milliseconds