Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/python/data/kernel_tests/
Dget_single_element_test.py81 def increment_fn(x): function
86 return dataset_ops.Dataset.range(1).map(increment_fn)
101 def increment_fn(x): function
110 return dataset_ops.Dataset.range(1).map(increment_fn)
Dreduce_test.py136 def increment_fn(x): function
141 return dataset_ops.Dataset.range(10).map(increment_fn)
Dcache_test.py303 def increment_fn(x): function
307 dataset = dataset_ops.Dataset.range(10).map(increment_fn).cache().repeat(2)
326 def increment_fn(x): function
330 dataset = dataset_ops.Dataset.range(10).map(increment_fn).cache()
Dmap_test.py508 def increment_fn(x): function
513 dataset = apply_map(dataset, increment_fn)
/external/tensorflow/tensorflow/compiler/tf2xla/
Dfunctionalize_control_flow_test.cc470 NodeDef increment_fn; in AddNoinlineFunctionToGraph() local
471 increment_fn.set_name(node_name); in AddNoinlineFunctionToGraph()
472 increment_fn.set_op("increment_fn"); in AddNoinlineFunctionToGraph()
473 *increment_fn.add_input() = "while/Identity"; in AddNoinlineFunctionToGraph()
474 *increment_fn.add_input() = "^while/Identity"; in AddNoinlineFunctionToGraph()
476 graph->AddNode(increment_fn, &status); in AddNoinlineFunctionToGraph()