Searched defs:axis_node (Results 1 – 8 of 8) sorted by relevance
47 auto* axis_node = graph_builder_->AddConstNodeWithData( in PopulateSubGraph() local
62 NodeDef* axis_node = in MakeGather() local
235 NodeDef axis_node = concat_node_match.inputs[2].node; in FuseBatchNormWithConvConcat() local
300 const NodeDef& axis_node = match.inputs[2].node; in SparsifyGatherInternal() local
424 auto* axis_node = context.graph_view->GetNode( in ReduceTransposerKeepDims() local491 auto* axis_node = context.graph_view->GetNode( in ReduceTransposerValidAxisNode() local2969 auto* axis_node = context.graph_view->GetNode( in TEST_F() local3402 auto* axis_node = context.graph_view->GetNode( in TEST_F() local3491 auto* axis_node = context.graph_view->GetNode( in TEST_F() local
1357 int axis_node = 0; in TransposeNode() local1510 auto* axis_node = regular_fanin_1.node_view(); in IsReduceAxisSupported() local
1963 auto axis_node = node_map_->GetNode(node_->input(1)); in IsAlongAxis() local
2444 NodeDef* axis_node = optimized_graph->add_node(); in SimplifyPack() local