Home
last modified time | relevance | path

Searched refs:MakeOpDef (Results 1 – 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/core/framework/
Dshape_inference_test.cc102 static OpDef MakeOpDef(int num_inputs, int num_outputs) { in MakeOpDef() function
117 InferenceContext c(kVersion, &def, MakeOpDef(1, 1), {Unknown()}, {}, {}, {}); in TEST_F()
132 InferenceContext c(kVersion, &def, MakeOpDef(1, 2), {S({1})}, {}, {}, {}); in TEST_F()
170 InferenceContext c(kVersion, &def, MakeOpDef(1, 2), {S({1, 2, 3})}, {}, {}, in TEST_F()
189 InferenceContext c(kVersion, &def, MakeOpDef(2, 2), in TEST_F()
211 InferenceContext c(kVersion, &def, MakeOpDef(2, 2), {S({3}), S({4})}, in TEST_F()
234 InferenceContext c(kVersion, &def, MakeOpDef(2, 2), {S({3}), S({4})}, in TEST_F()
257 InferenceContext c(kVersion, &def, MakeOpDef(3, 2), in TEST_F()
298 InferenceContext c(kVersion, &def, MakeOpDef(3, 2), in TEST_F()
312 InferenceContext c(kVersion, &def, MakeOpDef(2, 2), in TEST_F()
[all …]
Dcommon_shape_fns_test.cc37 OpDef MakeOpDef(int num_inputs, int num_outputs) { in MakeOpDef() function
1120 InferenceContext c(TF_GRAPH_DEF_VERSION, &def, MakeOpDef(3, 1), in TEST()
1133 InferenceContext c(TF_GRAPH_DEF_VERSION, &def, MakeOpDef(3, 1), in TEST()
1146 InferenceContext c(TF_GRAPH_DEF_VERSION, &def, MakeOpDef(3, 1), in TEST()
1160 InferenceContext c(TF_GRAPH_DEF_VERSION, &def, MakeOpDef(3, 1), in TEST()
1174 InferenceContext c(TF_GRAPH_DEF_VERSION, &def, MakeOpDef(3, 1), in TEST()
1188 InferenceContext c(TF_GRAPH_DEF_VERSION, &def, MakeOpDef(3, 1), in TEST()
1201 InferenceContext c(TF_GRAPH_DEF_VERSION, &def, MakeOpDef(3, 1), in TEST()
1214 InferenceContext c(TF_GRAPH_DEF_VERSION, &def, MakeOpDef(3, 1), in TEST()
1227 InferenceContext c(TF_GRAPH_DEF_VERSION, &def, MakeOpDef(3, 1), in TEST()
[all …]