| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_test_util.cc | 201 TFE_OpSetAttrType(op, "dtype", TF_FLOAT); in TestVariable()
220 TFE_OpSetAttrType(op, "dtype", TF_FLOAT); in TestVariable()
257 TFE_OpSetAttrType(op, "T", TFE_TensorHandleDataType(a)); in AddOp()
272 TFE_OpSetAttrType(op, "T", TFE_TensorHandleDataType(a)); in MatMulOp()
285 TFE_OpSetAttrType(op, "T", TFE_TensorHandleDataType(a)); in IdentityOp()
298 TFE_OpSetAttrType(op, "T", TFE_TensorHandleDataType(a)); in ShapeOp()
328 TFE_OpSetAttrType(op, "Tidx", TF_INT32); in MinOp()
330 TFE_OpSetAttrType(op, "T", TFE_TensorHandleDataType(input)); in MinOp()
|
| D | custom_device_test.cc | 116 TFE_OpSetAttrType(op.get(), "dtype", TF_FLOAT); in TEST()
135 TFE_OpSetAttrType(op.get(), "dtype", TF_FLOAT); in TEST()
150 TFE_OpSetAttrType(op.get(), "dtype", TF_FLOAT); in TEST()
201 TFE_OpSetAttrType(op.get(), "dtype", TF_FLOAT); in TEST()
220 TFE_OpSetAttrType(op.get(), "dtype", TF_FLOAT); in TEST()
236 TFE_OpSetAttrType(op.get(), "dtype", TF_FLOAT); in TEST()
|
| D | c_api_test.cc | 1214 TFE_OpSetAttrType(op, "dtype", TF_FLOAT); in TEST()
1259 TFE_OpSetAttrType(op, "dtype", TF_FLOAT); in BM_ReadVariable()
1318 TFE_OpSetAttrType(op, "T", TF_FLOAT); in TEST()
1380 TFE_OpSetAttrType(op, "T", TF_FLOAT); in TEST()
1674 TFE_OpSetAttrType(var_op, "dtype", TF_INT64); in TEST()
1679 TFE_OpSetAttrType(copy_op, "dtype", TF_FLOAT); in TEST()
1710 TFE_OpSetAttrType(var_op, "dtype", TF_INT64); in TEST()
1785 TFE_OpSetAttrType(original_var_op, "dtype", TF_INT64); in TEST()
|
| D | c_api.h | 313 TF_CAPI_EXPORT extern void TFE_OpSetAttrType(TFE_Op* op, const char* attr_name,
|
| D | c_api_cluster_test.cc | 97 TFE_OpSetAttrType(op, "dtype", TF_FLOAT); in ReadVariable()
|
| D | c_api.cc | 754 void TFE_OpSetAttrType(TFE_Op* op, const char* attr_name, TF_DataType value) { in TFE_OpSetAttrType() function
1033 TFE_OpSetAttrType(op, attr_name, in SetOpAttrValueScalar()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental_test.cc | 207 TFE_OpSetAttrType(reshape_op, "T", TF_FLOAT); in TEST_F()
208 TFE_OpSetAttrType(reshape_op, "Tshape", TF_INT32); in TEST_F()
218 TFE_OpSetAttrType(fill_op, "T", TF_FLOAT); in TEST_F()
219 TFE_OpSetAttrType(fill_op, "Tshape", TF_INT32); in TEST_F()
|
| /external/tensorflow/tensorflow/c/eager/parallel_device/ |
| D | parallel_device_testlib.cc | 40 TFE_OpSetAttrType(op.get(), "dtype", type); in Create()
88 TFE_OpSetAttrType(op.get(), "dtype", type_); in Read()
101 TFE_OpSetAttrType(op.get(), "dtype", type_); in GeneralAssignment()
|
| D | parallel_device_lib_test.cc | 58 TFE_OpSetAttrType(handle_op.get(), "dtype", TF_FLOAT); in TEST()
76 TFE_OpSetAttrType(read_op.get(), "dtype", TF_FLOAT); in TEST()
114 TFE_OpSetAttrType(handle_op.get(), "dtype", TF_FLOAT); in TEST()
|
| D | parallel_device_lib.h | 248 TFE_OpSetAttrType(const_op.get(), "dtype", datatype_enum); in ScalarsFromSequence()
|
| D | parallel_device_test.cc | 393 TFE_OpSetAttrType(op.get(), "T", TFE_TensorHandleDataType(input)); in CollectiveSum()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tensor.cc | 247 TFE_OpSetAttrType(op, "SrcT", src_type_enum); in EagerCast() local
248 TFE_OpSetAttrType(op, "DstT", dst_type_enum); in EagerCast()
|
| D | pywrap_tfe_src.cc | 661 TFE_OpSetAttrType(op, key, static_cast<TF_DataType>(value)); in SetOpAttrScalar()
3329 TFE_OpSetAttrType(op, "dtype", static_cast<TF_DataType>(value));
3443 TFE_OpSetAttrType(op, input_arg.type_attr().data(), dtype);
|