Home
last modified time | relevance | path

Searched refs:fn_op (Results 1 – 6 of 6) sorted by relevance

/external/tensorflow/tensorflow/python/eager/
Dfunction_argument_naming_test.py48 fn_op = fn.get_concrete_function(
53 [inp.op.name for inp in fn_op.inputs])
56 [inp.op.get_attr('_user_specified_name') for inp in fn_op.inputs])
57 self.assertEqual(2, len(fn_op.graph.structured_outputs))
60 fn_op(constant_op.constant(1.), constant_op.constant(2.)))
63 fn_op(a=constant_op.constant(1.), b=constant_op.constant(2.)))
72 fn_op = fn.get_concrete_function(
77 [inp.op.name for inp in fn_op.inputs])
80 [inp.op.get_attr('_user_specified_name') for inp in fn_op.inputs])
81 self.assertEqual(2, len(fn_op.graph.structured_outputs))
[all …]
Dfunction_test.py1016 fn_op = fn.get_concrete_function(x, x)
1017 self.assertEqual(fn_op.output_dtypes, None)
1018 self.assertEqual(fn_op.output_shapes, None)
1019 self.assertAllEqual(fn_op(x, x), None)
/external/tensorflow/tensorflow/c/eager/
Dunified_api_testutil.cc97 AbstractOperationPtr fn_op(ctx->CreateOperation()); in RunModel() local
98 TF_RETURN_IF_ERROR(fn_op->Reset(fn_name, /*raw_device_name=*/nullptr)); in RunModel()
100 TF_RETURN_IF_ERROR(fn_op->AddInput(input)); in RunModel()
104 TF_RETURN_IF_ERROR(fn_op->Execute( in RunModel()
Dc_api_unified_experimental_test.cc404 TF_AbstractOp* fn_op = TF_NewAbstractOp(eager_execution_ctx); in TEST_P() local
405 TF_AbstractOpSetOpType(fn_op, fn_name.c_str(), status.get()); in TEST_P()
417 TF_ExecuteOperation(fn_op, 1, &input_t, add_outputs, status.get()); in TEST_P()
431 TF_DeleteAbstractOp(fn_op); in TEST_P()
496 TF_AbstractOp* fn_op = TF_NewAbstractOp(eager_execution_ctx); in TEST_P() local
497 TF_AbstractOpSetOpType(fn_op, fn_name.c_str(), status.get()); in TEST_P()
516 TF_ExecuteOperation(fn_op, 1, &input_t, mm_outputs, status.get()); in TEST_P()
538 TF_DeleteAbstractOp(fn_op); in TEST_P()
652 TF_AbstractOp* fn_op = TF_NewAbstractOp(eager_execution_ctx); in TEST_P() local
653 TF_AbstractOpSetOpType(fn_op, fn_name.c_str(), s); in TEST_P()
[all …]
Dgradients_util.cc283 AbstractOperationPtr fn_op(ctx->CreateOperation()); in RunModel() local
284 TF_RETURN_IF_ERROR(fn_op->Reset(fn_name, /*raw_device_name=*/nullptr)); in RunModel()
286 TF_RETURN_IF_ERROR(fn_op->AddInput(input)); in RunModel()
290 TF_RETURN_IF_ERROR(fn_op->Execute( in RunModel()
/external/tensorflow/tensorflow/python/saved_model/
Dsave.py854 for fn_op in fn_graph.get_operations():
855 exported_operations.append((fn_name, fn_op))