Searched refs:TF_SetAttrType (Results 1 – 8 of 8) sorted by relevance
| /external/tensorflow/tensorflow/c/ |
| D | c_test_util.cc | 112 TF_SetAttrType(desc, "dtype", dtype); in PlaceholderHelper()
133 TF_SetAttrType(desc, "dtype", TF_TensorType(t)); in ConstHelper()
282 TF_SetAttrType(desc, "dtype", dtype); in RandomUniform()
294 TF_SetAttrType(desc, "T", TF_INT32); in Split3Helper()
|
| D | c_api.h | 336 TF_CAPI_EXPORT extern void TF_SetAttrType(TF_OperationDescription* desc,
|
| D | c_api_function_test.cc | 1284 TF_SetAttrType(desc, "dtype", TF_INT32); in NodeWithAttrHelper()
1732 TF_SetAttrType(desc, "Tcond", TF_BOOL); in TEST_F()
|
| D | c_api.cc | 882 void TF_SetAttrType(TF_OperationDescription* desc, const char* attr_name, in TF_SetAttrType() function
1823 TF_SetAttrType(desc, "dtype", TF_OperationOutputType(parent_input)); in CreateInput()
|
| D | c_api_test.cc | 1724 TF_SetAttrType(desc, "dtype", TF_FLOAT); in FloatConst2x2()
2113 TF_SetAttrType(desc, "v", TF_COMPLEX128); in TEST_F()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_experimental_test.cc | 303 TF_SetAttrType(arg_descr, "dtype", TF_INT32); in TEST()
309 TF_SetAttrType(id_descr, "T", TF_INT32); in TEST()
379 TF_SetAttrType(arg_descr, "dtype", TF_INT32); in TEST()
385 TF_SetAttrType(id_descr, "T", TF_INT32); in TEST()
|
| /external/tensorflow/tensorflow/go/ |
| D | graph.go | 407 C.TF_SetAttrType(cdesc, cAttrName, C.TF_DataType(value))
|
| /external/tensorflow/tensorflow/python/client/ |
| D | tf_session_helper.cc | 644 TF_SetAttrType(desc, "dtype", dtype); in CreatePlaceholder()
|