Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/grappler/optimizers/data/
Dmap_and_batch_fusion_test.cc36 std::vector<string> range_inputs(3); in TEST() local
37 range_inputs[0] = start_node->name(); in TEST()
38 range_inputs[1] = stop_node->name(); in TEST()
39 range_inputs[2] = step_node->name(); in TEST()
41 NodeDef *range_node = graph_utils::AddNode("", "RangeDataset", range_inputs, in TEST()
119 std::vector<string> range_inputs(3); in TEST() local
120 range_inputs[0] = start_node->name(); in TEST()
121 range_inputs[1] = stop_node->name(); in TEST()
122 range_inputs[2] = step_node->name(); in TEST()
124 NodeDef *range_node = graph_utils::AddNode("", "RangeDataset", range_inputs, in TEST()
[all …]
Dshuffle_and_repeat_fusion_test.cc44 std::vector<string> range_inputs(3); in TEST() local
45 range_inputs[0] = start_node->name(); in TEST()
46 range_inputs[1] = stop_node->name(); in TEST()
47 range_inputs[2] = step_node->name(); in TEST()
48 NodeDef *range_node = graph_utils::AddNode("", "RangeDataset", range_inputs, in TEST()
112 std::vector<string> range_inputs(3); in TEST() local
113 range_inputs[0] = start_node->name(); in TEST()
114 range_inputs[1] = stop_node->name(); in TEST()
115 range_inputs[2] = step_node->name(); in TEST()
116 NodeDef *range_node = graph_utils::AddNode("", "RangeDataset", range_inputs, in TEST()
Dnoop_elimination_test.cc67 std::vector<string> range_inputs = {start_node->name(), stop_node->name(), in MakeRangeNode() local
70 return graph_utils::AddNode("", "RangeDataset", range_inputs, in MakeRangeNode()
/external/tensorflow/tensorflow/core/kernels/data/
Ddataset_test_base.h94 gtl::InlinedVector<TensorValue, 4> range_inputs; in CreateRangeDataset() local
96 &range_inputs, range_kernel->input_types(), TensorShape({}), {start})); in CreateRangeDataset()
98 &range_inputs, range_kernel->input_types(), TensorShape({}), {end})); in CreateRangeDataset()
100 &range_inputs, range_kernel->input_types(), TensorShape({}), {step})); in CreateRangeDataset()
102 TF_RETURN_IF_ERROR(CreateOpKernelContext(range_kernel.get(), &range_inputs, in CreateRangeDataset()
104 TF_RETURN_IF_ERROR(CheckOpKernelInput(*range_kernel, range_inputs)); in CreateRangeDataset()