Searched refs:TF_OperationOutputType (Results 1 – 8 of 8) sorted by relevance
93 return DataType(C.TF_OperationOutputType(p.c()))
148 return static_cast<jint>(TF_OperationOutputType(TF_Output{op, output_index})); in Java_org_tensorflow_GraphOperation_dtype()
59 return static_cast<tensorflow::DataType>(TF_OperationOutputType(output_)); in DataType()
430 TF_CAPI_EXPORT extern TF_DataType TF_OperationOutputType(TF_Output oper_out);
1096 TF_DataType TF_OperationOutputType(TF_Output oper_out) { in TF_OperationOutputType() function1826 TF_SetAttrType(desc, "dtype", TF_OperationOutputType(parent_input)); in CreateInput()
472 EXPECT_EQ(TF_INT32, TF_OperationOutputType(TF_Output{feed, 0})); in TEST()505 EXPECT_EQ(TF_INT32, TF_OperationOutputType(TF_Output{add, 0})); in TEST()
633 m.def("TF_OperationOutputType", TF_OperationOutputType, in PYBIND11_MODULE()
2044 output_type = pywrap_tf_session.TF_OperationOutputType(tf_output)2210 int(pywrap_tf_session.TF_OperationOutputType(self._tf_output(i)))