| /external/tensorflow/tensorflow/lite/ |
| D | graph_info.cc | 121 bool UpdateNode(int node_index) { in UpdateNode() function in tflite::__anon4572c5030111::PartitionGraphIntoIndependentNodeSubsetsImpl
189 if (UpdateNode(node_index)) { in BuildNodeSubset()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer.cc | 693 Status LayoutSensitiveOpTransposer::UpdateNode(TransposeContext* context, in UpdateNode() function in tensorflow::grappler::LayoutSensitiveOpTransposer
747 TF_RETURN_IF_ERROR(UpdateNode(context, node)); in TransposeNode()
762 TF_RETURN_IF_ERROR(UpdateNode(context, node)); in TransposeNode()
789 TF_RETURN_IF_ERROR(UpdateNode(context, node)); in TransposeNode()
813 TF_RETURN_IF_ERROR(UpdateNode(context, node)); in TransposeNode()
832 TF_RETURN_IF_ERROR(UpdateNode(context, node)); in TransposeNode()
866 TF_RETURN_IF_ERROR(UpdateNode(context, node)); in TransposeNode()
888 TF_RETURN_IF_ERROR(UpdateNode(context, node)); in TransposeNode()
908 TF_RETURN_IF_ERROR(UpdateNode(context, node)); in TransposeNode()
931 TF_RETURN_IF_ERROR(UpdateNode(context, node)); in TransposeNode()
[all …]
|
| D | generic_layout_optimizer_transposer.h | 199 Status UpdateNode(TransposeContext* context, utils::MutableNodeView* node);
|
| D | generic_layout_optimizer_transposer_test.cc | 590 TEST_F(TransposerTest, UpdateNode) { in TEST_F() argument
604 TF_ASSERT_OK(transposer.UpdateNode(&context, conv2d)); in TEST_F()
649 TF_ASSERT_OK(transposer.UpdateNode(&context, conv2d)); in TEST_F()
|
| /external/tensorflow/tensorflow/core/grappler/ |
| D | mutable_graph_view.h | 83 Status UpdateNode(absl::string_view node_name, absl::string_view op,
|
| D | mutable_graph_view_test.cc | 226 graph.UpdateNode("bar_2", "IdentityN", kDevice, {{"T", list_value}})); in TEST()
251 TF_EXPECT_OK(graph.UpdateNode("bar_2", "NotImportant", kDevice, {})); in TEST()
273 TF_EXPECT_OK(graph.UpdateNode("foo", "Switch", kDevice, {})); in TEST()
292 Status s = graph.UpdateNode("foo", "Switch", kDevice, {{"T", attr}}); in TEST()
|
| D | mutable_graph_view.cc | 493 Status MutableGraphView::UpdateNode( in UpdateNode() function in tensorflow::grappler::MutableGraphView
|
| /external/tensorflow/tensorflow/core/common_runtime/ |
| D | shape_refiner.h | 110 Status UpdateNode(const Node* node, bool relax, bool* refined);
|
| D | shape_refiner_test.cc | 1344 TF_ASSERT_OK(m.UpdateNode(dequeue, false /* relax */, &refined)); in TEST_F()
1355 TF_ASSERT_OK(m.UpdateNode(dequeue, true /* relax */, &refined)); in TEST_F()
1366 TF_ASSERT_OK(m.UpdateNode(dequeue, true /* relax */, &refined)); in TEST_F()
1378 TF_ASSERT_OK(m.UpdateNode(dequeue, /*relax=*/false, &refined)); in TEST_F()
|
| D | shape_refiner.cc | 290 Status ShapeRefiner::UpdateNode(const Node* node, bool relax, bool* refined) { in UpdateNode() function in tensorflow::ShapeRefiner
|
| /external/tensorflow/tensorflow/core/grappler/costs/ |
| D | graph_properties.cc | 960 Status UpdateNode(const NodeDef* node, bool* refined) { in UpdateNode() function in tensorflow::grappler::SymbolicShapeRefiner
2273 TF_RETURN_IF_ERROR(shape_refiner->UpdateNode(node, new_shapes)); in UpdateEnter()
2318 TF_RETURN_IF_ERROR(shape_refiner->UpdateNode(n, new_shapes)); in UpdateShapes()
2390 return shape_refiner->UpdateNode(queue_node, new_shapes); in UpdateQueue()
2398 return shape_refiner->UpdateNode(queue_node, new_shapes); in UpdateQueue()
2421 return shape_refiner->UpdateNode(queue_node, &dummy_new_shapes); in UpdateQueue()
|
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| D | graph_transferer.cc | 357 shape_refiner->UpdateNode(input_node, false /* relax */, &refined)); in TransformGraphToAddAggregatedInputNode()
402 shape_refiner->UpdateNode(created_node, false /* relax */, &refined)); in TransformGraphToAddAggregatedInputNode()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 913 shape_refiner_->UpdateNode(node, /*relax=*/false, &inferred); in AddNodesToShapeRefiner()
|