Searched refs:TFE_OpSetAttrShape (Results 1 – 9 of 9) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_lib_test.cc | 59 TFE_OpSetAttrShape(handle_op.get(), "shape", /*dims=*/nullptr, /*num_dims=*/0, in TEST()
115 TFE_OpSetAttrShape(handle_op.get(), "shape", /*dims=*/nullptr, /*num_dims=*/0, in TEST()
|
| D | parallel_device_testlib.cc | 41 TFE_OpSetAttrShape(op.get(), "shape", dims, num_dims, status); in Create()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.h | 318 TF_CAPI_EXPORT extern void TFE_OpSetAttrShape(TFE_Op* op, const char* attr_name,
|
| D | custom_device_test.cc | 117 TFE_OpSetAttrShape(op.get(), "shape", {}, 0, status.get()); in TEST()
202 TFE_OpSetAttrShape(op.get(), "shape", {}, 0, status.get()); in TEST()
|
| D | c_api.cc | 762 void TFE_OpSetAttrShape(TFE_Op* op, const char* attr_name, const int64_t* dims, in TFE_OpSetAttrShape() function
1039 TFE_OpSetAttrShape(op, attr_name, nullptr, -1, status); in SetOpAttrValueScalar()
1046 TFE_OpSetAttrShape(op, attr_name, dims.get(), num_dims, status); in SetOpAttrValueScalar()
|
| D | c_api_test_util.cc | 202 TFE_OpSetAttrShape(op, "shape", {}, 0, status); in TestVariable()
|
| D | c_api_test.cc | 1675 TFE_OpSetAttrShape(var_op, "shape", {}, 0, status); in TEST()
1711 TFE_OpSetAttrShape(var_op, "shape", {}, 0, status); in TEST()
1786 TFE_OpSetAttrShape(original_var_op, "shape", {}, 0, status); in TEST()
|
| /external/tensorflow/tensorflow/java/src/main/native/ |
| D | eager_operation_builder_jni.cc | 290 TFE_OpSetAttrShape(op, cname, cvalue.get(), static_cast<int>(num_dims), in Java_org_tensorflow_EagerOperationBuilder_setAttrShape()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tfe_src.cc | 664 TFE_OpSetAttrShape(op, key, nullptr, -1, status); in SetOpAttrScalar()
676 TFE_OpSetAttrShape(op, key, nullptr, -1, status); in SetOpAttrScalar()
690 TFE_OpSetAttrShape(op, key, dims.get(), num_dims, status); in SetOpAttrScalar()
|