| 2392 | } |
| 2393 | |
| 2394 | void AddPlaceholder(const string& name, ArrayDataType type, |
| 2395 | GraphDef* tensorflow_graph) { |
| 2396 | tensorflow::NodeDef* placeholder = tensorflow_graph->add_node(); |
| 2397 | placeholder->set_op("Placeholder"); |
| 2398 | switch (type) { |
| 2399 | case ArrayDataType::kBool: |
| 2400 | (*placeholder->mutable_attr())["dtype"].set_type(DT_BOOL); |
| 2401 | break; |
| 2402 | case ArrayDataType::kFloat: |
| 2403 | (*placeholder->mutable_attr())["dtype"].set_type(DT_FLOAT); |
| 2404 | break; |
| 2405 | case ArrayDataType::kUint8: |
| 2406 | (*placeholder->mutable_attr())["dtype"].set_type(DT_UINT8); |
| 2407 | break; |
| 2408 | case ArrayDataType::kInt32: |
| 2409 | (*placeholder->mutable_attr())["dtype"].set_type(DT_INT32); |
| 2410 | break; |
| 2411 | case ArrayDataType::kInt64: |
| 2412 | (*placeholder->mutable_attr())["dtype"].set_type(DT_INT64); |
| 2413 | break; |
| 2414 | case ArrayDataType::kInt16: |
| 2415 | (*placeholder->mutable_attr())["dtype"].set_type(DT_INT16); |
| 2416 | break; |
| 2417 | case ArrayDataType::kComplex64: |
| 2418 | (*placeholder->mutable_attr())["dtype"].set_type(DT_COMPLEX64); |
| 2419 | break; |
| 2420 | default: |
| 2421 | LOG(FATAL) << "Unexpected data type in array \"" << name << "\""; |
| 2422 | } |
| 2423 | placeholder->set_name(name); |
| 2424 | } |
| 2425 | |
| 2426 | void AddPlaceholderForRNNState(const Model& model, const string& name, int size, |
| 2427 | GraphDef* tensorflow_graph) { |
no test coverage detected