Searched refs:TF_SetAttrShape (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | graph_operation_builder_jni.cc | 267 TF_SetAttrShape(d, cname, cvalue.get(), static_cast<int>(num_dims)); in Java_org_tensorflow_GraphOperationBuilder_setAttrShape()
|
| /external/tensorflow/tensorflow/go/ |
| D | graph.go | 473 C.TF_SetAttrShape(cdesc, cAttrName, dimsp, ndims)
|
| /external/tensorflow/tensorflow/c/ |
| D | c_test_util.cc | 114 TF_SetAttrShape(desc, "shape", dims.data(), dims.size()); in PlaceholderHelper()
|
| D | c_api.h | 357 TF_CAPI_EXPORT extern void TF_SetAttrShape(TF_OperationDescription* desc,
|
| D | c_api_test.cc | 2213 TF_SetAttrShape(desc, "v", nullptr, -1); in TEST_F()
2224 TF_SetAttrShape(desc, "v", partial_shape, sz); in TEST_F()
|
| D | c_api.cc | 905 void TF_SetAttrShape(TF_OperationDescription* desc, const char* attr_name, in TF_SetAttrShape() function
|