Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/common_runtime/
Dfunction_testlib.cc71 TF_CHECK_OK(scope->DoShapeInference(n)); in Call()
Dconstant_folding_test.cc344 TF_ASSERT_OK(s.DoShapeInference(times_two)); in TEST_F()
378 TF_ASSERT_OK(s.DoShapeInference(non_cpu)); in TEST_F()
/external/tensorflow/tensorflow/cc/framework/
Dscope.h208 Status DoShapeInference(Node* node) const;
Dscope.cc426 Status Scope::DoShapeInference(Node* node) const { in DoShapeInference() function in tensorflow::Scope
/external/tensorflow/tensorflow/cc/ops/
Dconst_op.cc37 scope.UpdateStatus(scope.DoShapeInference(ret)); in ConstHelper()
Dconst_op.h62 scope.UpdateStatus(scope.DoShapeInference(ret)); in Const()
Dwhile_loop.cc85 TF_RETURN_IF_ERROR(scope.DoShapeInference(merge_node)); in CreateMerge()
/external/tensorflow/tensorflow/core/framework/
Dvariant_op_copy_test.cc144 scope.UpdateStatus(scope.DoShapeInference(ret)); in CreateTestVariant()
/external/tensorflow/tensorflow/compiler/tf2xla/
Dxla_compiler_test.cc116 scope.UpdateStatus(scope.DoShapeInference(ret)); in DummyReadResourceCC()
645 TF_ASSERT_OK(scope.DoShapeInference(fill)); in TEST_F()
/external/tensorflow/tensorflow/core/graph/
Dgraph_partition_test.cc178 scope.UpdateStatus(scope.DoShapeInference(ret)); in ConstructOp()