Searched refs:FindFunctionNodeWithOp (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/grappler/optimizers/data/ |
| D | vectorization_utils_test.cc | 253 function_utils::FindFunctionNodeWithOp("MapDefun", *vectorized)); in TEST()
258 function_utils::FindFunctionNodeWithOp("Cast", *vectorized)); in TEST()
327 function_utils::FindFunctionNodeWithOp("Cast", *vectorized)); in TEST()
408 function_utils::FindFunctionNodeWithOp("Cast", *vectorized)); in TEST()
411 function_utils::FindFunctionNodeWithOp("Unpack", *vectorized)); in TEST()
484 function_utils::FindFunctionNodeWithOp("Print", *map_defun_fn)); in TEST()
486 function_utils::FindFunctionNodeWithOp("Cast", *map_defun_fn)); in TEST()
563 function_utils::FindFunctionNodeWithOp("Const", *vectorized)); in TEST()
565 function_utils::FindFunctionNodeWithOp("Cast", *vectorized)); in TEST()
652 function_utils::FindFunctionNodeWithOp("Cast", *vectorized)); in TEST()
[all …]
|
| D | function_utils_test.cc | 91 TEST(FunctionUtilsTest, FindFunctionNodeWithOp) { in TEST() argument
94 FindFunctionNodeWithOp("weird_op_that_should_not_be_there", function), in TEST()
96 EXPECT_NE(FindFunctionNodeWithOp("Mul", function), -1); in TEST()
|
| D | function_utils.cc | 131 return FindFunctionNodeWithOp(op, function) != -1; in ContainsFunctionNodeWithOp()
157 int FindFunctionNodeWithOp(StringPiece op, const FunctionDef& function) { in FindFunctionNodeWithOp() function
|
| D | function_utils.h | 97 int FindFunctionNodeWithOp(StringPiece op, const FunctionDef& function);
|
| D | fusion_utils_test.cc | 116 function_utils::FindFunctionNodeWithOp("Equal", *fused_function)); in TEST()
|
| D | hoist_random_uniform.cc | 164 function_utils::FindFunctionNodeWithOp("RandomUniform", in MakeLessStatefulFunction()
|
| D | map_vectorization_test.cc | 226 branch_0->node_def(function_utils::FindFunctionNodeWithOp( in CheckVectorized()
|