Searched refs:AttrTypeMapForOp (Results 1 – 7 of 7) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | attr_builder_test.cc | 41 Status s = AttrTypeMapForOp("SomeFunctionName", &m, &is_function); in TEST()
50 s = AttrTypeMapForOp("MatMul", &m, &is_function); in TEST()
64 s = AttrTypeMapForOp("Squeeze", &m, &is_function); in TEST()
|
| D | attr_builder.h | 50 Status AttrTypeMapForOp(const char* op_name, const AttrTypeMap** out,
|
| D | attr_builder.cc | 65 Status AttrTypeMapForOp(const char* op_name, const AttrTypeMap** out, in AttrTypeMapForOp() function
|
| D | execute.cc | 1060 tensorflow::AttrTypeMapForOp("_Send", &types, &is_function)); in ExecuteSend()
1091 tensorflow::AttrTypeMapForOp("_Recv", &types, &is_function)); in ExecuteRecv()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/eager/ |
| D | eager_service_impl.cc | 158 TF_RETURN_IF_ERROR(tensorflow::AttrTypeMapForOp(name, &types, &is_function)); in ExecuteOp()
|
| /external/tensorflow/tensorflow/lite/delegates/flex/ |
| D | kernel.cc | 241 tensorflow::AttrTypeMapForOp(name_.c_str(), &attr_types, &is_function), in BuildEagerOp()
|
| /external/tensorflow/tensorflow/c/eager/ |
| D | c_api.cc | 567 status->status = tensorflow::AttrTypeMapForOp(name, &types, &is_function); in TFE_NewOp()
|