Home
last modified time | relevance | path

Searched refs:NumRegularFanins (Results 1 – 10 of 10) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer_test.cc76 ASSERT_GT(node->NumRegularFanins(), port); in VerifyRegularFaninMatch()
420 ASSERT_EQ(updated_sum_node->NumRegularFanins(), 2); in ReduceTransposerKeepDims()
427 ASSERT_EQ(axis_node->NumRegularFanins(), 1); in ReduceTransposerKeepDims()
436 ASSERT_EQ(z_output_node->NumRegularFanins(), 1); in ReduceTransposerKeepDims()
487 ASSERT_EQ(updated_max_node->NumRegularFanins(), 2); in ReduceTransposerValidAxisNode()
494 ASSERT_EQ(axis_node->NumRegularFanins(), 1); in ReduceTransposerValidAxisNode()
499 ASSERT_EQ(z_output_node->NumRegularFanins(), 1); in ReduceTransposerValidAxisNode()
701 ASSERT_EQ(transpose_node1->NumRegularFanins(), 2); in TEST_F()
707 ASSERT_EQ(transpose_node2->NumRegularFanins(), 2); in TEST_F()
713 ASSERT_EQ(updated_fbng->NumRegularFanins(), 5); in TEST_F()
[all …]
Dgeneric_layout_optimizer_test.cc203 ASSERT_GE(node->NumRegularFanins(), port); in VerifyRegularFaninMatch()
252 ASSERT_EQ(input_transpose_node->NumRegularFanins(), 2); in TEST_F()
257 ASSERT_EQ(conv2d_node->NumRegularFanins(), 2); in TEST_F()
266 ASSERT_EQ(output_transpose_node->NumRegularFanins(), 2); in TEST_F()
271 ASSERT_EQ(output_node->NumRegularFanins(), 1); in TEST_F()
410 ASSERT_EQ(node_i2_output->NumRegularFanins(), 1); in TEST_F()
432 ASSERT_EQ(conv2d_backprop_node->NumRegularFanins(), 3); in TEST_F()
444 ASSERT_EQ(input_sizes_node->NumRegularFanins(), 1); in TEST_F()
478 ASSERT_EQ(conv2d_node->NumRegularFanins(), 2); in TEST_F()
487 ASSERT_EQ(shape_node->NumRegularFanins(), 1); in TEST_F()
[all …]
Dremapper.cc413 if (node_view->NumRegularFanins() < 1) return false; in IsCpuCompatibleDataType()
453 if (node_view->NumRegularFanins() < 1) return false; in IsCpuCompatibleDataType()
501 if (node_view->NumRegularFanins() < 1) return false; in IsCpuCompatibleDataType()
521 if (squeeze_node_view->NumRegularFanins() < 1) return false; in IsCpuCompatibleDataType()
571 if (node_view->NumRegularFanins() < 1) return false; in IsCpuCompatibleDataType()
605 if (node_view->NumRegularFanins() < 1) return false; in IsCpuCompatibleDataType()
638 if (add_node_view.NumRegularFanins() < port_id + 1) return false; in IsCpuCompatibleDataType()
672 if (HasControlFaninOrFanout(node_view) || node_view.NumRegularFanins() != 2) in IsCpuCompatibleDataType()
743 if (node_view->NumRegularFanins() < 1) return false; in IsCpuCompatibleDataType()
906 if (node_view->NumRegularFanins() < 1) return false; in IsCpuCompatibleDataType()
[all …]
Dgeneric_layout_optimizer.cc235 if (node->NumRegularFanins() < 1) { in EraseCancellableNodes()
257 if (node->NumRegularFanins() > 1) { in EraseCancellableNodes()
261 if (fanin_node->NumRegularFanins() > 1) { in EraseCancellableNodes()
Dimplementation_selector.cc112 const int last_input_index = backward_node->NumRegularFanins() - 1; in FindForwardNode()
Dgeneric_layout_optimizer_transposer.cc110 const int num_regular_fanins = node.NumRegularFanins(); in GetRegularFaninPorts()
590 if (port < node.NumRegularFanins() && port >= 0) { in GetFaninPortRank()
605 if (port < node.NumRegularFanins() && port >= 0) { in IsFaninPortDimsNIfConst()
949 if (node->NumRegularFanins() == 6) { in TransposeNode()
1138 const int num_regular_fanins = node.NumRegularFanins(); in GetVariadicNDFaninPorts()
2129 if (node.NumRegularFanins() > 0) { in GetDataFaninPorts()
/external/tensorflow/tensorflow/core/grappler/utils/
Dgraph_view.cc571 fanin_id.node(), pos - node_view.NumRegularFanins()); in AddFaninsInternal()
731 const int last_index = fanout_view->NumRegularFanins() - in RemovedOrMissingNodeFanoutsWellFormed()
1146 const int num_regular_fanins = node_view->NumRegularFanins(); in RemoveControllingFaninInternal()
1220 node_view.NumRegularFanins() - diff.num_regular_inputs_to_remove; in ApplyNodeUpdates()
1221 for (int i = first_index; i < node_view.NumRegularFanins(); ++i) { in ApplyNodeUpdates()
1226 node_view.NumRegularFanins(), diff.num_regular_inputs_to_remove); in ApplyNodeUpdates()
1231 int curr_index = node_view.NumRegularFanins(); in ApplyNodeUpdates()
1248 curr_control_start != node_view.NumRegularFanins()) { in ApplyNodeUpdates()
1250 node_def->mutable_input()->begin() + node_view.NumRegularFanins(), in ApplyNodeUpdates()
1295 const int num_regular_fanins = node_view->NumRegularFanins(); in RemoveAllFaninFanoutInternal()
[all …]
Dgraph_view_internal.h208 int NumRegularFanins() const { return regular_fanins_.size(); } in NumRegularFanins() function
496 const int num_regular_fanins = node_view->NumRegularFanins(); in AddOrUpdateRegularFanin()
538 const int num_regular_fanins = node_view->NumRegularFanins(); in RemoveRegularFanin()
716 node_view->NumRegularFanins() - diff->num_regular_inputs_to_remove - 1; in IsWellFormed()
Dframe.cc41 if (node.NumRegularFanins() + node.NumControllingFanins() == 0) { in InferFromGraphViewT()
Dgraph_view_test.cc222 EXPECT_EQ(d_node->NumRegularFanins(), 2); in TEST()
561 TYPED_TEST(TypedNodeViewTest, NumRegularFanins) { in TYPED_TEST() argument
577 EXPECT_EQ(a_node->NumRegularFanins(), 4); in TYPED_TEST()
578 EXPECT_EQ(b_node->NumRegularFanins(), 2); in TYPED_TEST()
579 EXPECT_EQ(c_node->NumRegularFanins(), 0); in TYPED_TEST()
580 EXPECT_EQ(d_node->NumRegularFanins(), 0); in TYPED_TEST()
818 expected_node_view.NumRegularFanins(); in CompareGraphViewWithGraph()
820 node_view->NumRegularFanins() == expected_num_regular_fanins; in CompareGraphViewWithGraph()
830 if (i < node_view->NumRegularFanins()) { in CompareGraphViewWithGraph()