Searched defs:batch_size_node (Results 1 – 5 of 5) sorted by relevance
62 NodeDef *batch_size_node = graph_utils::AddScalarConstNode<int64>(5, &graph); in TEST() local144 NodeDef *batch_size_node = graph_utils::AddScalarConstNode<int64>(5, &graph); in TEST() local230 NodeDef *batch_size_node = graph_utils::AddScalarConstNode<int64>(5, &graph); in TEST() local315 NodeDef *batch_size_node = graph_utils::AddScalarConstNode<int64>(5, &graph); in TEST() local380 NodeDef *batch_size_node = graph_utils::AddScalarConstNode<int64>(5, &graph); in TEST() local
134 auto batch_size_node = graph_utils::AddScalarConstNode(batch_size, graph); in AddBatchNode() local345 auto batch_size_node = graph_utils::AddScalarConstNode(batch_size, graph); in AddMapAndBatchNode() local
125 auto batch_size_node = GetInputNode(node, graph, arg_index); in GetBatchSizeFromBatchNode() local
139 Node* batch_size_node; in AsGraphDefInternal() local
153 Node* batch_size_node; in AsGraphDefInternal() local