Searched refs:TFE_OpSetAttrType (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api_test_util.cc | 126 TFE_OpSetAttrType(op, "T", TFE_TensorHandleDataType(a)); in MatMulOp()
139 TFE_OpSetAttrType(op, "T", TFE_TensorHandleDataType(a)); in ShapeOp()
169 TFE_OpSetAttrType(op, "Tidx", TF_INT32); in MinOp()
171 TFE_OpSetAttrType(op, "T", TFE_TensorHandleDataType(input)); in MinOp()
|
| D | c_api.h | 315 TF_CAPI_EXPORT extern void TFE_OpSetAttrType(TFE_Op* op, const char* attr_name,
|
| D | c_api_test.cc | 1426 TFE_OpSetAttrType(op, "dtype", TF_FLOAT); in CreateVariable()
1441 TFE_OpSetAttrType(op, "dtype", TF_FLOAT); in CreateVariable()
1480 TFE_OpSetAttrType(op, "dtype", TF_FLOAT); in TEST()
1521 TFE_OpSetAttrType(op, "dtype", TF_FLOAT); in BM_ReadVariable()
1584 TFE_OpSetAttrType(op, "T", TF_FLOAT); in TEST()
|
| D | c_api.cc | 672 void TFE_OpSetAttrType(TFE_Op* op, const char* attr_name, TF_DataType value) { in TFE_OpSetAttrType() function
937 TFE_OpSetAttrType(op, attr_name, in SetOpAttrValueScalar()
|
| /external/tensorflow/tensorflow/c/ |
| D | c_api_experimental_test.cc | 238 TFE_OpSetAttrType(send_op, "T", TF_FLOAT); in TEST()
254 TFE_OpSetAttrType(recv_op, "tensor_type", TF_FLOAT); in TEST()
427 TFE_OpSetAttrType(squeeze, "T", TF_INT32); in TEST_F()
|
| /external/tensorflow/tensorflow/python/eager/ |
| D | pywrap_tensor.cc | 193 TFE_OpSetAttrType(op, "SrcT", src_type_enum); in EagerCast() local
194 TFE_OpSetAttrType(op, "DstT", dst_type_enum); in EagerCast()
|
| D | pywrap_tfe_src.cc | 552 TFE_OpSetAttrType(op, key, static_cast<TF_DataType>(value)); in SetOpAttrScalar()
2226 TFE_OpSetAttrType(op, "dtype", static_cast<TF_DataType>(value));
2383 TFE_OpSetAttrType(op, input_arg.type_attr().data(), dtype);
|