Home
last modified time | relevance | path

Searched refs:batch_size_node (Results 1 – 6 of 6) 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()
145 NodeDef *batch_size_node = graph_utils::AddScalarConstNode<int64>(5, &graph); in TEST() local
152 batch_inputs[1] = batch_size_node->name(); in TEST()
232 NodeDef *batch_size_node = graph_utils::AddScalarConstNode<int64>(5, &graph); in TEST() local
237 batch_inputs[1] = batch_size_node->name(); in TEST()
298 NodeDef *batch_size_node = graph_utils::AddScalarConstNode<int64>(5, &graph); in TEST() local
301 batch_inputs[1] = batch_size_node->name(); in TEST()
Drebatch.cc145 NodeDef* batch_size_node = graph_utils::GetInputNode(node, *graph, 1); in MutateBatchSize() local
148 if (batch_size_node->op() != kConstOp) { in MutateBatchSize()
150 batch_size_node->op(), " instead."); in MutateBatchSize()
154 GetNodeAttr(*batch_size_node, "value", &batch_size_tensor)); in MutateBatchSize()
Dmap_vectorization_test.cc121 auto batch_size_node = graph_utils::AddScalarConstNode(batch_size, graph); in AddBatchNode() local
129 {input_dataset, batch_size_node->name(), drop_remainder->name()}, in AddBatchNode()
136 /*inputs=*/{input_dataset, batch_size_node->name()}, in AddBatchNode()
288 auto batch_size_node = graph_utils::AddScalarConstNode(batch_size, graph); in AddMapAndBatchNode() local
297 {input_dataset, batch_size_node->name(), in AddMapAndBatchNode()
/external/tensorflow/tensorflow/core/kernels/data/experimental/
Ddense_to_sparse_batch_dataset_op.cc131 Node* batch_size_node; in AsGraphDefInternal() local
132 TF_RETURN_IF_ERROR(b->AddScalar(batch_size_, &batch_size_node)); in AsGraphDefInternal()
141 this, {input_node, batch_size_node, row_shape_node}, output)); in AsGraphDefInternal()
Dmap_and_batch_dataset_op.cc206 Node* batch_size_node; in AsGraphDefInternal() local
207 TF_RETURN_IF_ERROR(b->AddScalar(batch_size_, &batch_size_node)); in AsGraphDefInternal()
240 std::make_pair(2, batch_size_node), in AsGraphDefInternal()
Dnuma_map_and_batch_dataset_op.cc157 Node* batch_size_node; in AsGraphDefInternal() local
158 TF_RETURN_IF_ERROR(b->AddScalar(batch_size_, &batch_size_node)); in AsGraphDefInternal()
189 std::make_pair(2, batch_size_node), in AsGraphDefInternal()