Searched refs:graph_op (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental_test.cc | 348 TF_Operation* graph_op = in AddEagerOpToGraphAndCheck() local
351 CHECK_NOTNULL(graph_op); in AddEagerOpToGraphAndCheck()
353 checker(graph_op); in AddEagerOpToGraphAndCheck()
404 AddEagerOpToGraphAndCheck(op, [this, &axis](TF_Operation* graph_op) { in TEST_F() argument
406 TF_OperationGetAttrBool(graph_op, "keep_dims", &value, status_); in TEST_F()
410 TF_OperationGetAttrType(graph_op, "Tidx", &dtype, status_); in TEST_F()
413 TF_OperationGetAttrType(graph_op, "T", &dtype, status_); in TEST_F()
460 identityn, [this, kNumInputs](TF_Operation* graph_op) { in TEST_F() argument
461 EXPECT_EQ(TF_OperationNumInputs(graph_op), kNumInputs); in TEST_F()
462 EXPECT_EQ(TF_OperationInputListLength(graph_op, "input", status_), in TEST_F()
[all …]
|
| D | c_api_experimental.cc | 9103 auto* graph_op = TF_FinishOperation(desc, status); in TFE_AddEagerOpToGraph() local
9107 *num_retvals = TF_OperationNumOutputs(graph_op); in TFE_AddEagerOpToGraph()
9110 auto output = TF_Output{graph_op, i}; in TFE_AddEagerOpToGraph()
9114 return graph_op; in TFE_AddEagerOpToGraph()
|
| /external/tensorflow/tensorflow/python/ops/ |
| D | resource_variable_ops.py | 51 def get_resource_handle_data(graph_op): argument
52 assert type(graph_op) == ops.Tensor # pylint: disable=unidiomatic-typecheck
55 graph_op.graph._c_graph, graph_op._as_tf_output()) # pylint: disable=protected-access
|