Home
last modified time | relevance | path

Searched defs:batch_size_node (Results 1 – 5 of 5) sorted by relevance

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmap_and_batch_fusion_test.cc62 NodeDef *batch_size_node = graph_utils::AddScalarConstNode<int64>(5, &graph); in TEST() local
144 NodeDef *batch_size_node = graph_utils::AddScalarConstNode<int64>(5, &graph); in TEST() local
230 NodeDef *batch_size_node = graph_utils::AddScalarConstNode<int64>(5, &graph); in TEST() local
315 NodeDef *batch_size_node = graph_utils::AddScalarConstNode<int64>(5, &graph); in TEST() local
380 NodeDef *batch_size_node = graph_utils::AddScalarConstNode<int64>(5, &graph); in TEST() local
Dmap_vectorization_test.cc134 auto batch_size_node = graph_utils::AddScalarConstNode(batch_size, graph); in AddBatchNode() local
345 auto batch_size_node = graph_utils::AddScalarConstNode(batch_size, graph); in AddMapAndBatchNode() local
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dcompute_batch_size_op.cc125 auto batch_size_node = GetInputNode(node, graph, arg_index); in GetBatchSizeFromBatchNode() local
Ddense_to_sparse_batch_dataset_op.cc139 Node* batch_size_node; in AsGraphDefInternal() local
Dmap_and_batch_dataset_op.cc153 Node* batch_size_node; in AsGraphDefInternal() local