Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dlower_if_while_test.cc157 TF_ASSERT_OK(root.DoShapeInference(while_node)); in TEST()
263 TF_ASSERT_OK(root.DoShapeInference(if_node)); in TEST()
328 TF_ASSERT_OK(root.DoShapeInference(const_node)); in TEST()
Dlower_if_op_test.cc79 TF_ASSERT_OK(root.DoShapeInference(written_if)); in TEST()
203 TF_ASSERT_OK(root.DoShapeInference(if_node)); in TEST()
285 TF_ASSERT_OK(root.DoShapeInference(written_if)); in TEST()
Dlower_while_op_test.cc72 TF_ASSERT_OK(root.DoShapeInference(while_node)); in TEST()
174 TF_ASSERT_OK(root.DoShapeInference(while_node)); in TEST()
277 TF_ASSERT_OK(root.DoShapeInference(while_node)); in TEST()
Dfunction_testlib.cc127 TF_CHECK_OK(scope->DoShapeInference(n)); in Call()
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()
/external/tensorflow/tensorflow/lite/tools/accuracy/
Drun_tflite_model_stage.cc40 s.UpdateStatus(s.DoShapeInference(ret)); in AddToGraph()
/external/tensorflow/tensorflow/compiler/jit/
Dbuild_xla_ops_pass_test.cc182 TF_ASSERT_OK(root.DoShapeInference(call)); in TEST_F()
217 TF_ASSERT_OK(root.DoShapeInference(call)); in TEST_F()
Dencapsulate_xla_computations_pass_test.cc66 TF_CHECK_OK(scope.DoShapeInference(launch)); in MakeOuterGraph()
/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/cc/ops/
Dconst_op.h62 scope.UpdateStatus(scope.DoShapeInference(ret)); in Const()
Dconst_op.cc37 scope.UpdateStatus(scope.DoShapeInference(ret)); in ConstHelper()
Dwhile_loop.cc85 TF_RETURN_IF_ERROR(scope.DoShapeInference(merge_node)); in CreateMerge()
/external/tensorflow/tensorflow/core/framework/
Dvariant_op_copy_test.cc145 scope.UpdateStatus(scope.DoShapeInference(ret)); in CreateTestVariant()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler_test.cc122 scope.UpdateStatus(scope.DoShapeInference(ret)); in DummyReadResourceCC()
916 TF_ASSERT_OK(scope.DoShapeInference(fill)); in TEST_F()
1359 TF_ASSERT_OK(scope.DoShapeInference(fill)); in TEST_F()
/external/tensorflow/tensorflow/core/graph/
Dgraph_partition_test.cc179 scope.UpdateStatus(scope.DoShapeInference(ret)); in ConstructOp()