| /external/tensorflow/tensorflow/lite/ |
| D | graph_info.cc | 131 bool UpdateNode(int node_index) { in UpdateNode() function in tflite::__anon1add66280111::PartitionGraphIntoIndependentNodeSubsetsImpl
214 if (UpdateNode(node_index)) { in BuildNodeSubset()
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| D | generic_layout_optimizer_transposer.cc | 688 Status LayoutSensitiveOpTransposer::UpdateNode(TransposeContext* context, in UpdateNode() function in tensorflow::grappler::LayoutSensitiveOpTransposer
742 TF_RETURN_IF_ERROR(UpdateNode(context, node)); in TransposeNode()
757 TF_RETURN_IF_ERROR(UpdateNode(context, node)); in TransposeNode()
784 TF_RETURN_IF_ERROR(UpdateNode(context, node)); in TransposeNode()
808 TF_RETURN_IF_ERROR(UpdateNode(context, node)); in TransposeNode()
827 TF_RETURN_IF_ERROR(UpdateNode(context, node)); in TransposeNode()
861 TF_RETURN_IF_ERROR(UpdateNode(context, node)); in TransposeNode()
883 TF_RETURN_IF_ERROR(UpdateNode(context, node)); in TransposeNode()
903 TF_RETURN_IF_ERROR(UpdateNode(context, node)); in TransposeNode()
926 TF_RETURN_IF_ERROR(UpdateNode(context, node)); in TransposeNode()
[all …]
|
| D | generic_layout_optimizer_transposer.h | 206 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 | 515 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 | 1392 TF_ASSERT_OK(m.UpdateNode(dequeue, false /* relax */, &refined)); in TEST_F()
1403 TF_ASSERT_OK(m.UpdateNode(dequeue, true /* relax */, &refined)); in TEST_F()
1414 TF_ASSERT_OK(m.UpdateNode(dequeue, true /* relax */, &refined)); in TEST_F()
1426 TF_ASSERT_OK(m.UpdateNode(dequeue, /*relax=*/false, &refined)); in TEST_F()
|
| D | shape_refiner.cc | 306 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 | 989 Status UpdateNode(const NodeDef* node, bool* refined) { in UpdateNode() function in tensorflow::grappler::SymbolicShapeRefiner
2317 TF_RETURN_IF_ERROR(shape_refiner->UpdateNode(node, new_shapes)); in UpdateEnter()
2362 TF_RETURN_IF_ERROR(shape_refiner->UpdateNode(n, new_shapes)); in UpdateShapes()
2434 return shape_refiner->UpdateNode(queue_node, new_shapes); in UpdateQueue()
2442 return shape_refiner->UpdateNode(queue_node, new_shapes); in UpdateQueue()
2465 return shape_refiner->UpdateNode(queue_node, &dummy_new_shapes); in UpdateQueue()
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/translate/ |
| D | import_model.cc | 1023 shape_refiner_->UpdateNode(node, /*relax=*/false, &inferred); in AddNodesToShapeRefiner()
|