Searched refs:TF_SetAttrType (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/c/ |
| D | c_test_util.cc | 87 TF_SetAttrType(desc, "dtype", TF_INT32); in PlaceholderHelper()
104 TF_SetAttrType(desc, "dtype", TF_TensorType(t)); in ConstHelper()
234 TF_SetAttrType(desc, "dtype", dtype); in RandomUniform()
246 TF_SetAttrType(desc, "T", TF_INT32); in Split3Helper()
|
| D | c_api.h | 511 TF_CAPI_EXPORT extern void TF_SetAttrType(TF_OperationDescription* desc,
|
| D | c_api_test.cc | 1642 TF_SetAttrType(desc, "dtype", TF_FLOAT); in FloatConst2x2()
1912 TF_SetAttrType(desc, "v", TF_COMPLEX128); in TEST_F()
|
| D | c_api.cc | 1203 void TF_SetAttrType(TF_OperationDescription* desc, const char* attr_name, in TF_SetAttrType() function
2114 TF_SetAttrType(desc, "dtype", TF_OperationOutputType(parent_input)); in CreateInput()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_test.cc | 635 TF_SetAttrType(arg_descr, "dtype", TF_INT32); in TEST()
641 TF_SetAttrType(id_descr, "T", TF_INT32); in TEST()
697 TF_SetAttrType(arg_descr, "dtype", TF_INT32); in TEST()
703 TF_SetAttrType(id_descr, "T", TF_INT32); in TEST()
|
| /external/tensorflow/tensorflow/go/ |
| D | graph.go | 305 C.TF_SetAttrType(cdesc, cAttrName, C.TF_DataType(value))
|
| /external/tensorflow/tensorflow/contrib/tensorrt/segment/ |
| D | segment_test.cc | 73 TF_SetAttrType(desc, "dtype", TF_INT32); in PlaceholderHelper()
|