Searched refs:axis_arg (Results 1 – 2 of 2) sorted by relevance
54 const Tidx axis_arg = in Compute() local56 const Tidx axis = (axis_arg < 0) ? input.dims() + axis_arg : axis_arg; in Compute()
3158 int axis_arg = -1; in PartialConcatConstFolding() local3163 axis_arg = 0; in PartialConcatConstFolding()3166 axis_arg = num_non_control_inputs - 1; in PartialConcatConstFolding()3172 node_map_->GetNode(NodeName(node->input(axis_arg))); in PartialConcatConstFolding()3231 added_node->add_input(node->input(axis_arg)); in PartialConcatConstFolding()3234 node_map_->AddOutput(NodeName(node->input(axis_arg)), added_node->name()); in PartialConcatConstFolding()