Home
last modified time | relevance | path

Searched refs:increment_fn (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/python/data/experimental/kernel_tests/
Dget_single_element_test.py79 def increment_fn(x): function
84 return dataset_ops.Dataset.range(1).map(increment_fn)
98 def increment_fn(x): function
107 return dataset_ops.Dataset.range(1).map(increment_fn)
/external/tensorflow/tensorflow/python/data/kernel_tests/
Dreduce_test.py131 def increment_fn(x): function
136 return dataset_ops.Dataset.range(10).map(increment_fn)
Dmap_test.py440 def increment_fn(x): function
444 dataset = dataset_ops.Dataset.range(10).map(increment_fn)
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow_test.cc343 NodeDef increment_fn; in AddNoinlineFunctionToGraph() local
344 increment_fn.set_name(node_name); in AddNoinlineFunctionToGraph()
345 increment_fn.set_op("increment_fn"); in AddNoinlineFunctionToGraph()
346 *increment_fn.add_input() = "while/Identity"; in AddNoinlineFunctionToGraph()
347 *increment_fn.add_input() = "^while/Identity"; in AddNoinlineFunctionToGraph()
349 graph->AddNode(increment_fn, &status); in AddNoinlineFunctionToGraph()