Searched refs:DEFINE_GET_ATTR (Results 1 – 2 of 2) sorted by relevance
| /external/tensorflow/tensorflow/core/common_runtime/eager/ |
| D | attr_builder.cc | 138 #define DEFINE_GET_ATTR(TYPE, FIELD, ATTR_TYPE) \ macro
152 DEFINE_GET_ATTR(float, f, "float");
153 DEFINE_GET_ATTR(int, i, "int");
154 DEFINE_GET_ATTR(bool, b, "bool");
155 DEFINE_GET_ATTR(tensorflow::DataType, type, "type");
157 #undef DEFINE_GET_ATTR
|
| /external/tensorflow/tensorflow/core/framework/ |
| D | node_def_util.cc | 208 #define DEFINE_GET_ATTR(TYPE, FIELD, ATTR_TYPE, APPEND_OP, CAST, ...) \ macro
265 DEFINE_GET_ATTR(tstring, s, "string", emplace_back, v, ;)
267 DEFINE_GET_ATTR(string, s, "string", emplace_back, v, ;)
269 DEFINE_GET_ATTR(int64, i, "int", emplace_back, v, ;)
271 DEFINE_GET_ATTR(
288 DEFINE_GET_ATTR(float, f, "float", emplace_back, v, ;)
293 DEFINE_GET_ATTR(bool, b, "bool", push_back, v, ;)
295 DEFINE_GET_ATTR(DataType, type, "type", emplace_back, static_cast<DataType>(v),
300 DEFINE_GET_ATTR(TensorShapeProto, shape, "shape", emplace_back, v, ;)
301 DEFINE_GET_ATTR(TensorShape, shape, "shape", emplace_back, TensorShape(v),
[all …]
|