Searched refs:TFE_OpSetAttrShape (Results 1 – 4 of 4) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.h | 199 TF_CAPI_EXPORT extern void TFE_OpSetAttrShape(TFE_Op* op, const char* attr_name,
|
| D | c_api.cc | 361 void TFE_OpSetAttrShape(TFE_Op* op, const char* attr_name, const int64_t* dims, in TFE_OpSetAttrShape() function
|
| D | c_api_test.cc | 871 TFE_OpSetAttrShape(op, "shape", {}, 0, status); in CreateVariable()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tfe_src.cc | 383 TFE_OpSetAttrShape(op, key, nullptr, -1, status); in SetOpAttrScalar()
395 TFE_OpSetAttrShape(op, key, nullptr, -1, status); in SetOpAttrScalar()
407 TFE_OpSetAttrShape(op, key, dims.get(), num_dims, status); in SetOpAttrScalar()
475 TFE_OpSetAttrShape(op, attr_name, nullptr, -1, status); in SetOpAttrScalarDefault()
482 TFE_OpSetAttrShape(op, attr_name, dims.get(), num_dims, status); in SetOpAttrScalarDefault()
|