Home
last modified time | relevance | path

Searched refs: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
67 batch_inputs[1] = batch_size_node->name(); in TEST()
144 NodeDef *batch_size_node = graph_utils::AddScalarConstNode<int64>(5, &graph); in TEST() local
151 batch_inputs[1] = batch_size_node->name(); in TEST()
230 NodeDef *batch_size_node = graph_utils::AddScalarConstNode<int64>(5, &graph); in TEST() local
235 batch_inputs[1] = batch_size_node->name(); in TEST()
315 NodeDef *batch_size_node = graph_utils::AddScalarConstNode<int64>(5, &graph); in TEST() local
320 batch_inputs[1] = batch_size_node->name(); in TEST()
380 NodeDef *batch_size_node = graph_utils::AddScalarConstNode<int64>(5, &graph); in TEST() local
383 batch_inputs[1] = batch_size_node->name(); in TEST()
Dmap_vectorization_test.cc134 auto batch_size_node = graph_utils::AddScalarConstNode(batch_size, graph); in AddBatchNode() local
142 {input_dataset, batch_size_node->name(), drop_remainder->name()}, in AddBatchNode()
149 /*inputs=*/{input_dataset, batch_size_node->name()}, in AddBatchNode()
345 auto batch_size_node = graph_utils::AddScalarConstNode(batch_size, graph); in AddMapAndBatchNode() local
354 {input_dataset, batch_size_node->name(), in AddMapAndBatchNode()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Dcompute_batch_size_op.cc125 auto batch_size_node = GetInputNode(node, graph, arg_index); in GetBatchSizeFromBatchNode() local
127 auto s = GetScalarConstNodeValue(*batch_size_node, &batch_size); in GetBatchSizeFromBatchNode()
Ddense_to_sparse_batch_dataset_op.cc139 Node* batch_size_node; in AsGraphDefInternal() local
140 TF_RETURN_IF_ERROR(b->AddScalar(batch_size_, &batch_size_node)); in AsGraphDefInternal()
149 this, {input_node, batch_size_node, row_shape_node}, output)); in AsGraphDefInternal()
Dmap_and_batch_dataset_op.cc153 Node* batch_size_node; in AsGraphDefInternal() local
154 TF_RETURN_IF_ERROR(b->AddScalar(batch_size_, &batch_size_node)); in AsGraphDefInternal()
174 std::make_pair(2, batch_size_node), in AsGraphDefInternal()