Searched refs:new_shapes (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.h | 144 SymbolicShapeRefiner* shape_refiner, bool* new_shapes);
149 bool* new_shapes);
154 bool* new_shapes) const;
157 const NodeDef* node, bool* new_shapes);
163 const NodeDef* n, bool* new_shapes) const;
167 SymbolicShapeRefiner* shape_refiner, TopoQueue* new_shapes,
|
| D | graph_properties.cc | 1785 bool* new_shapes) const { in UpdateMerge()
1791 *new_shapes = true; in UpdateMerge()
1826 if (*new_shapes || !shape_refiner->EquivalentShapes(out, ic->output(0))) { in UpdateMerge()
1829 *new_shapes = true; in UpdateMerge()
1837 const NodeDef* node, bool* new_shapes) { in UpdateEnter() argument
1840 TF_RETURN_IF_ERROR(shape_refiner->UpdateNode(node, new_shapes)); in UpdateEnter()
1852 *new_shapes = true; in UpdateEnter()
1858 *new_shapes = true; in UpdateEnter()
1866 const NodeDef* n, bool* new_shapes) const { in UpdateShapes()
1870 TF_RETURN_IF_ERROR(UpdateEnter(shape_refiner, n, new_shapes)); in UpdateShapes()
[all …]
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | sort_simplifier.cc | 70 std::vector<Shape> new_shapes; in RemoveUnusedOperandFromSort() local
74 new_shapes.push_back(sort->operand(i)->shape()); in RemoveUnusedOperandFromSort()
78 Shape new_sort_shape = new_shapes.size() == 1 in RemoveUnusedOperandFromSort()
79 ? new_shapes[0] in RemoveUnusedOperandFromSort()
80 : ShapeUtil::MakeTupleShape(new_shapes); in RemoveUnusedOperandFromSort()
119 new_shapes[new_index], new_sort, new_index)); in RemoveUnusedOperandFromSort()
|
| D | stable_sort_expander.cc | 101 std::vector<Shape> new_shapes = sort->operand_count() == 1 in ExpandInstruction()
104 new_shapes.push_back(iota_shape); in ExpandInstruction()
105 Shape new_sort_shape = ShapeUtil::MakeTupleShape(new_shapes); in ExpandInstruction()
|