Searched refs:TFE_OpSetAttrInt (Results 1 – 6 of 6) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_test.cc | 369 TFE_OpSetAttrInt(op.get(), "num_replicas", 1); in TEST()
394 TFE_OpSetAttrInt(op.get(), "group_size", group_size); in CollectiveSum()
395 TFE_OpSetAttrInt(op.get(), "group_key", 0); in CollectiveSum()
396 TFE_OpSetAttrInt(op.get(), "instance_key", 0); in CollectiveSum()
|
| D | parallel_device_testlib.h | 124 TFE_OpSetAttrInt(op.get(), "N", num_replicas); in CreatePerDeviceValues()
|
| D | parallel_device_testlib.cc | 165 TFE_OpSetAttrInt(op.get(), "num_replicas", num_replicas); in ExtractPerDeviceValues()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.h | 307 TF_CAPI_EXPORT extern void TFE_OpSetAttrInt(TFE_Op* op, const char* attr_name,
|
| D | c_api.cc | 732 void TFE_OpSetAttrInt(TFE_Op* op, const char* attr_name, int64_t value) { in TFE_OpSetAttrInt() function
1024 TFE_OpSetAttrInt(op, attr_name, static_cast<int64_t>(default_value.i())); in SetOpAttrValueScalar()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tfe_src.cc | 645 TFE_OpSetAttrInt(op, key, value); in SetOpAttrScalar()
3741 TFE_OpSetAttrInt(op, input_arg.number_attr().data(), len);
|