Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/costs/
Dgraph_properties.h175 SymbolicShapeRefiner* shape_refiner, bool* new_shapes);
180 bool* new_shapes);
185 bool* new_shapes) const;
188 const NodeDef* node, bool* new_shapes);
194 const NodeDef* n, bool* new_shapes) const;
198 SymbolicShapeRefiner* shape_refiner, TopoQueue* new_shapes,
Dgraph_properties.cc2262 bool* new_shapes) const { in UpdateMerge()
2268 *new_shapes = true; in UpdateMerge()
2303 if (*new_shapes || !shape_refiner->EquivalentShapes(out, ic->output(0))) { in UpdateMerge()
2306 *new_shapes = true; in UpdateMerge()
2314 const NodeDef* node, bool* new_shapes) { in UpdateEnter() argument
2317 TF_RETURN_IF_ERROR(shape_refiner->UpdateNode(node, new_shapes)); in UpdateEnter()
2329 *new_shapes = true; in UpdateEnter()
2335 *new_shapes = true; in UpdateEnter()
2343 const NodeDef* n, bool* new_shapes) const { in UpdateShapes()
2347 TF_RETURN_IF_ERROR(UpdateEnter(shape_refiner, n, new_shapes)); in UpdateShapes()
[all …]
/external/tensorflow/tensorflow/compiler/xla/service/
Dsort_simplifier.cc70 std::vector<const 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::MakeTupleShapeWithPtrs(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()
/external/pytorch/torch/onnx/_internal/exporter/
D_capture_strategies.py130 new_shapes = torch.export.dynamic_shapes.refine_dynamic_shapes_from_suggested_fixes(
137 model, args, kwargs=kwargs, dynamic_shapes=new_shapes
171 new_shapes = torch.export.dynamic_shapes.refine_dynamic_shapes_from_suggested_fixes(
178 model, args, kwargs=kwargs, dynamic_shapes=new_shapes, strict=False
/external/pytorch/test/export/
Dtest_export.py3237 new_shapes = refine_dynamic_shapes_from_suggested_fixes(
3240 export(Foo(), inps, dynamic_shapes=new_shapes)
3241 return new_shapes
3263 new_shapes = helper(Foo(), inps, dynamic_shapes)
3264 self.assertEqual(new_shapes["x"][0], 4)
3265 self.assertEqual(new_shapes["z"][0], 6)
3275 new_shapes = helper(Foo(), inps, dynamic_shapes)
3276 self.assertEqual(new_shapes["x"][0].min, 6)
3277 self.assertEqual(new_shapes["y"][0].max, 16)
3291 new_shapes = helper(Foo(), inps, dynamic_shapes)
[all …]