Home
last modified time | relevance | path

Searched refs:DoShapeInference (Results 1 – 20 of 20) sorted by relevance

/external/tensorflow/tensorflow/core/common_runtime/
Dlower_function_call_op_test.cc101 TF_ASSERT_OK(root.DoShapeInference(function_call)); in TEST()
177 TF_ASSERT_OK(root.DoShapeInference(tpu_call)); in TEST()
178 TF_ASSERT_OK(root.DoShapeInference(xla_call)); in TEST()
Dlower_if_op_test.cc88 TF_ASSERT_OK(root.DoShapeInference(written_if)); in TEST()
213 TF_ASSERT_OK(root.DoShapeInference(if_node)); in TEST()
295 TF_ASSERT_OK(root.DoShapeInference(written_if)); in TEST()
Dlower_functional_ops_test.cc169 TF_ASSERT_OK(root.DoShapeInference(while_node)); in TEST()
275 TF_ASSERT_OK(root.DoShapeInference(if_node)); in TEST()
Dlower_case_op_test.cc92 TF_ASSERT_OK(root.DoShapeInference(written_if)); in TEST()
249 TF_ASSERT_OK(root.DoShapeInference(case_node)); in TEST()
361 TF_ASSERT_OK(root.DoShapeInference(written_case)); in TEST()
Dfunction_testlib.cc127 TF_CHECK_OK(scope->DoShapeInference(n)); in Call()
Dlower_while_op_test.cc86 TF_ASSERT_OK(root.DoShapeInference(while_node)); in TEST()
433 TF_ASSERT_OK(root.DoShapeInference(while_node)); in TEST()
548 TF_ASSERT_OK(root.DoShapeInference(while_node)); in TEST()
Dconstant_folding_test.cc357 TF_ASSERT_OK(s.DoShapeInference(times_two)); in TEST_F()
391 TF_ASSERT_OK(s.DoShapeInference(non_cpu)); in TEST_F()
Dexecutor_test.cc638 TF_ASSERT_OK(root.DoShapeInference(while_node)); in BM_WhileLoopHelper()
/external/tensorflow/tensorflow/cc/ops/
Dconst_op.cc37 scope.UpdateStatus(scope.DoShapeInference(ret)); in ConstHelper()
Dconst_op.h64 scope.UpdateStatus(scope.DoShapeInference(ret)); in Const()
Dwhile_loop.cc85 TF_RETURN_IF_ERROR(scope.DoShapeInference(merge_node)); in CreateMerge()
/external/tensorflow/tensorflow/cc/framework/
Dscope.h223 Status DoShapeInference(Node* node) const;
Dscope.cc501 Status Scope::DoShapeInference(Node* node) const { in DoShapeInference() function in tensorflow::Scope
/external/tensorflow/tensorflow/compiler/jit/
Dbuild_xla_ops_pass_test.cc185 TF_ASSERT_OK(root.DoShapeInference(call)); in TEST_F()
222 TF_ASSERT_OK(root.DoShapeInference(call)); in TEST_F()
Dencapsulate_xla_computations_pass_test.cc66 TF_CHECK_OK(scope.DoShapeInference(launch)); in MakeOuterGraph()
Ddeadness_analysis_test.cc1239 scope.UpdateStatus(scope.DoShapeInference(ret)); in CreateSwitchN()
/external/tensorflow/tensorflow/core/framework/
Dvariant_op_copy_test.cc145 scope.UpdateStatus(scope.DoShapeInference(ret)); in CreateTestVariant()
/external/tensorflow/tensorflow/core/graph/
Dgraph_partition_test.cc179 scope.UpdateStatus(scope.DoShapeInference(ret)); in ConstructOp()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler_test.cc130 scope.UpdateStatus(scope.DoShapeInference(ret)); in DummyReadResourceCC()
982 TF_ASSERT_OK(scope.DoShapeInference(fill)); in TEST_F()
1428 TF_ASSERT_OK(scope.DoShapeInference(fill)); in TEST_F()
/external/tensorflow/tensorflow/core/grappler/optimizers/
Dgeneric_layout_optimizer_transposer_test.cc2765 scope.UpdateStatus(scope.DoShapeInference(ret)); in CreateConcatV1Op()