Searched refs:new_batch_node (Results 1 – 1 of 1) sorted by relevance
462 NodeDef new_batch_node; in ConvertMapAndBatch() local463 new_batch_node.set_op(kBatchV2Op); in ConvertMapAndBatch()465 &new_batch_node); in ConvertMapAndBatch()466 new_batch_node.add_input(new_map_node.name()); in ConvertMapAndBatch()467 new_batch_node.add_input(orig_node.input(num_map_inputs)); // batch_size in ConvertMapAndBatch()468 new_batch_node.add_input( in ConvertMapAndBatch()470 graph_utils::CopyShapesAndTypesAttrs(orig_node, &new_batch_node); in ConvertMapAndBatch()473 NodeDef* graph_batch_node = graph->AddNode(std::move(new_batch_node)); in ConvertMapAndBatch()