Searched refs:TFE_OpSetAttrShape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.h | 320 TF_CAPI_EXPORT extern void TFE_OpSetAttrShape(TFE_Op* op, const char* attr_name,
|
| D | c_api.cc | 677 void TFE_OpSetAttrShape(TFE_Op* op, const char* attr_name, const int64_t* dims, in TFE_OpSetAttrShape() function
943 TFE_OpSetAttrShape(op, attr_name, nullptr, -1, status); in SetOpAttrValueScalar()
950 TFE_OpSetAttrShape(op, attr_name, dims.get(), num_dims, status); in SetOpAttrValueScalar()
|
| D | c_api_test.cc | 1427 TFE_OpSetAttrShape(op, "shape", {}, 0, status); in CreateVariable()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tfe_src.cc | 555 TFE_OpSetAttrShape(op, key, nullptr, -1, status); in SetOpAttrScalar()
567 TFE_OpSetAttrShape(op, key, nullptr, -1, status); in SetOpAttrScalar()
581 TFE_OpSetAttrShape(op, key, dims.get(), num_dims, status); in SetOpAttrScalar()
|