Searched refs:TFE_OpSetAttrInt (Results 1 – 5 of 5) sorted by relevance
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental_test.cc | 244 TFE_OpSetAttrInt(send_op, "send_device_incarnation", 1234); in TEST()
259 TFE_OpSetAttrInt(recv_op, "send_device_incarnation", 1234); in TEST()
|
| D | c_api_experimental.cc | 8577 TFE_OpSetAttrInt(queueOp.get(), "capacity", 1); in createTFEQueue()
8613 TFE_OpSetAttrInt(op, "timeout_ms", -1); in createTFEEnqueue()
8634 TFE_OpSetAttrInt(op, "timeout_ms", -1); in createTFEDequeue()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.h | 309 TF_CAPI_EXPORT extern void TFE_OpSetAttrInt(TFE_Op* op, const char* attr_name,
|
| D | c_api.cc | 660 void TFE_OpSetAttrInt(TFE_Op* op, const char* attr_name, int64_t value) { in TFE_OpSetAttrInt() function
928 TFE_OpSetAttrInt(op, attr_name, static_cast<int64_t>(default_value.i())); in SetOpAttrValueScalar()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tfe_src.cc | 536 TFE_OpSetAttrInt(op, key, value); in SetOpAttrScalar()
2666 TFE_OpSetAttrInt(op, input_arg.number_attr().data(), len);
|