Home
last modified time | relevance | path

Searched refs:dt_vec (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/hexagon/
Dhexagon_ops_definitions.cc288 const string& op_type, const DataTypeVector& dt_vec, in EmplaceOpType() argument
295 std::forward_as_tuple(dt_vec, supported_op_type)); in EmplaceOpType()
386 const DataTypeVector& dt_vec) const { in GetOpIdFor()
392 if (dt_vec.empty()) { in GetOpIdFor()
401 if (std::get<0>(data_type_to_op) == dt_vec) { in GetOpIdFor()
Dhexagon_ops_definitions.h43 const string& op_type, const DataTypeVector& dt_vec,
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils.cc856 std::vector<DataType> dt_vec; in BuildClusterSubgraphDef() local
858 GetOutputTensorShapeType(*node_def, &dt_vec, &shape_vec).IgnoreError(); in BuildClusterSubgraphDef()
860 dt_vec.empty() ? DT_FLOAT : dt_vec.at(subgraph_input_port); in BuildClusterSubgraphDef()
1417 std::vector<DataType> dt_vec; in BuildNodeMapFromOpsDefinitions() local
1420 GetOutputTensorShapeType(node_def, &dt_vec, &shape_vec); in BuildNodeMapFromOpsDefinitions()
1425 node_def.op(), DataTypeVector(dt_vec.begin(), dt_vec.end())) != in BuildNodeMapFromOpsDefinitions()