Searched refs:axis_node_name (Results 1 – 2 of 2) sorted by relevance
405 axis_node_name = self._node.name + "/axis"409 output_axis_node.name = axis_node_name421 [self._node.input[0], self._node.input[1], axis_node_name])
2423 const string axis_node_name = OptimizedNodeName(*node, "_const_axis"); in SimplifyPack() local2425 node_map_->NodeExists(axis_node_name)) { in SimplifyPack()2441 !CreateNodeDef(axis_node_name, TensorValue(&axis_t), &new_node).ok()) { in SimplifyPack()2446 axis_node->set_name(axis_node_name); in SimplifyPack()