Searched refs:TF_SetAttrType (Results 1 – 9 of 9) 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 | 345 TF_CAPI_EXPORT extern void TF_SetAttrType(TF_OperationDescription* desc,
|
| D | c_api.cc | 884 void TF_SetAttrType(TF_OperationDescription* desc, const char* attr_name, in TF_SetAttrType() function
1826 TF_SetAttrType(desc, "dtype", TF_OperationOutputType(parent_input)); in CreateInput()
|
| D | c_api_test.cc | 1761 TF_SetAttrType(desc, "dtype", TF_FLOAT); in FloatConst2x2()
2150 TF_SetAttrType(desc, "v", TF_COMPLEX128); in TEST_F()
|
| D | c_api_function_test.cc | 1284 TF_SetAttrType(desc, "dtype", TF_INT32); in NodeWithAttrHelper()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_experimental_test.cc | 249 TF_SetAttrType(arg_descr, "dtype", TF_INT32); in TEST()
255 TF_SetAttrType(id_descr, "T", TF_INT32); in TEST()
|
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_test.cc | 476 TF_SetAttrType(placeholder_desc, "dtype", TF_FLOAT); in RegisterCollectiveMulFunction()
483 TF_SetAttrType(reduce_desc, "T", TF_FLOAT); in RegisterCollectiveMulFunction()
|
| /external/tensorflow/tensorflow/go/ |
| D | graph.go | 437 C.TF_SetAttrType(cdesc, cAttrName, C.TF_DataType(value))
|
| /external/tensorflow/tensorflow/python/client/ |
| D | tf_session_helper.cc | 639 TF_SetAttrType(desc, "dtype", dtype); in CreatePlaceholder()
|