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 | 545 TF_CAPI_EXPORT extern void TF_SetAttrType(TF_OperationDescription* desc,
|
| D | c_api_test.cc | 1723 TF_SetAttrType(desc, "dtype", TF_FLOAT); in FloatConst2x2()
2112 TF_SetAttrType(desc, "v", TF_COMPLEX128); in TEST_F()
|
| D | c_api.cc | 1306 void TF_SetAttrType(TF_OperationDescription* desc, const char* attr_name, in TF_SetAttrType() function
2233 TF_SetAttrType(desc, "dtype", TF_OperationOutputType(parent_input)); in CreateInput()
|
| D | c_api_function_test.cc | 1695 TF_SetAttrType(desc, "Tcond", TF_BOOL); in TEST_F()
|
| D | c_api_experimental.cc | 9036 TF_SetAttrType(desc, "dtype", in getOrCreateSymbolicTensor()
|
| /external/tensorflow/tensorflow/go/ |
| D | graph.go | 414 C.TF_SetAttrType(cdesc, cAttrName, C.TF_DataType(value))
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_test.cc | 1172 TF_SetAttrType(arg_descr, "dtype", TF_INT32); in TEST()
1178 TF_SetAttrType(id_descr, "T", TF_INT32); in TEST()
1239 TF_SetAttrType(arg_descr, "dtype", TF_INT32); in TEST()
1245 TF_SetAttrType(id_descr, "T", TF_INT32); in TEST()
|