Home
last modified time | relevance | path

Searched refs:new_shapes (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.h174 SymbolicShapeRefiner* shape_refiner, bool* new_shapes);
179 bool* new_shapes);
184 bool* new_shapes) const;
187 const NodeDef* node, bool* new_shapes);
193 const NodeDef* n, bool* new_shapes) const;
197 SymbolicShapeRefiner* shape_refiner, TopoQueue* new_shapes,
Dgraph_properties.cc2218 bool* new_shapes) const { in UpdateMerge()
2224 *new_shapes = true; in UpdateMerge()
2259 if (*new_shapes || !shape_refiner->EquivalentShapes(out, ic->output(0))) { in UpdateMerge()
2262 *new_shapes = true; in UpdateMerge()
2270 const NodeDef* node, bool* new_shapes) { in UpdateEnter() argument
2273 TF_RETURN_IF_ERROR(shape_refiner->UpdateNode(node, new_shapes)); in UpdateEnter()
2285 *new_shapes = true; in UpdateEnter()
2291 *new_shapes = true; in UpdateEnter()
2299 const NodeDef* n, bool* new_shapes) const { in UpdateShapes()
2303 TF_RETURN_IF_ERROR(UpdateEnter(shape_refiner, n, new_shapes)); in UpdateShapes()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dsort_simplifier.cc70 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()
Dstable_sort_expander.cc101 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()