| /external/tensorflow/tensorflow/contrib/hvx/hvx_ops_support_checker/ |
| hvx_ops_support_checker_main.cc | 64 std::vector<DataType> data_types; local
67 node_def, &data_types, &shapes);
68 if (data_types.empty() || shapes.empty()) {
71 CHECK_EQ(data_types.size(), shapes.size());
72 for (int i = 0; i < data_types.size(); ++i) {
73 LOG(INFO) << " Output(" << i << "): " << DataType_Name(data_types.at(i))
|
| /external/tensorflow/tensorflow/core/kernels/ |
| remote_fused_graph_execute_utils_test.cc | 381 std::vector<DataType> data_types; local
384 &data_types));
385 ASSERT_EQ(1, data_types.size());
386 EXPECT_EQ(DT_FLOAT, data_types.at(0));
400 std::vector<DataType> data_types; local
403 &data_types));
404 ASSERT_EQ(1, data_types.size());
405 EXPECT_EQ(DT_FLOAT, data_types.at(0));
[all...] |
| remote_fused_graph_execute_utils.cc | 367 const std::vector<DataType>& data_types,
369 AddNodeAttr(ATTR_OUTPUT_DATA_TYPES, data_types, node_def);
384 std::vector<DataType> data_types; local
390 data_types.emplace(data_types.begin(), tst->first);
396 AddOutputTensorShapeType(data_types, shapes, node_def);
401 AttrSlice attrs, std::vector<DataType>* data_types,
404 if (data_types != nullptr) {
405 status = GetNodeAttr(attrs, ATTR_OUTPUT_DATA_TYPES, data_types);
412 if (status.ok() && data_types != nullptr)
423 std::vector<DataType> data_types; local
[all...] |
| /external/tensorflow/tensorflow/core/kernels/hexagon/ |
| graph_transferer.cc | 175 std::vector<DataType> data_types; local
178 node->attrs(), &data_types, &shapes);
180 CHECK(data_types.size() > port);
181 graph_output_node_info.set_dtype(data_types.at(port));
329 std::vector<DataType> data_types; local
337 data_types.emplace_back(input_node_info_list.at(i).second.dtype());
348 data_types)
380 node_name, AGGREGATED_INPUT_NODE_NAME, i, data_types.at(i), graph,
383 std::vector<DataType> data_types; local
386 original_input_node->attrs(), &data_types, &shapes)
911 std::vector<DataType> data_types; local
[all...] |
