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
180 Node* axis_node = graph->create(prim::Constant); in CreateQuantizedWeights() 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
1352 int axis_node = 0; in TransposeNode() local1505 auto* axis_node = regular_fanin_1.node_view(); in IsReduceAxisSupported() local
2494 NodeDef* axis_node = optimized_graph->add_node(); in SimplifyPack() local