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()
|