Home
last modified time | relevance | path

Searched refs:GetOpIdFor (Results 1 – 8 of 8) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
Di_remote_fused_graph_ops_definitions.h37 virtual int GetOpIdFor(const string& op_type,
Dremote_fused_graph_execute_utils.cc1422 if (ops_definitions.GetOpIdFor( in BuildNodeMapFromOpsDefinitions()
/external/tensorflow/tensorflow/core/kernels/hexagon/
Dhexagon_ops_definitions.h34 int GetOpIdFor(const string& op_type, const DataTypeVector& dt) const final;
Dgraph_transferer.cc444 } else if (ops_definitions.GetOpIdFor(node.type_string(), {}) != in RegisterNode()
665 const int op_type_id = ops_definitions.GetOpIdFor(node.type_string(), {}); in RegisterNodeWithPaddingAndStrides()
694 const int op_type_id = ops_definitions.GetOpIdFor(node.type_string(), {}); in RegisterNodeWithRank()
720 const int op_type_id = ops_definitions.GetOpIdFor(node.type_string(), {}); in RegisterPadNode()
792 const int op_type_id = ops_definitions.GetOpIdFor("INPUT", {}); in RegisterInputNode()
811 const int op_type_id = ops_definitions.GetOpIdFor(op_type, {}); in RegisterFlattenNode()
827 const int op_type_id = ops_definitions.GetOpIdFor(node.type_string(), {}); in RegisterGenericNode()
Dhexagon_ops_definitions.cc385 int HexagonOpsDefinitions::GetOpIdFor(const string& op_type, in GetOpIdFor() function in tensorflow::HexagonOpsDefinitions
Dhexagon_control_wrapper.cc165 HexagonOpsDefinitions::getInstance().GetOpIdFor(OUTPUT_OP_NAME, {})); in SetupGraph()
Dgraph_transferer_test.cc56 int GetOpIdFor(const string& op_type, const DataTypeVector&) const final { in GetOpIdFor() function in tensorflow::TestGraphTransferOpsDefinitions
/external/tensorflow/tensorflow/contrib/hvx/hvx_ops_support_checker/
Dhvx_ops_support_checker_main.cc130 const int op_id = ops_definition.GetOpIdFor(node.op(), {}); in CheckOpsSupport()