Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/core/kernels/hexagon/
Dgraph_transferer.h60 const IRemoteFusedGraphOpsDefinitions& ops_definitions,
70 const IRemoteFusedGraphOpsDefinitions& ops_definitions,
119 const IRemoteFusedGraphOpsDefinitions& ops_definitions,
147 const IRemoteFusedGraphOpsDefinitions& ops_definitions,
151 const IRemoteFusedGraphOpsDefinitions& ops_definitions,
154 void RegisterPadNode(const IRemoteFusedGraphOpsDefinitions& ops_definitions,
157 void RegisterInputNode(const IRemoteFusedGraphOpsDefinitions& ops_definitions,
161 const IRemoteFusedGraphOpsDefinitions& ops_definitions,
165 const IRemoteFusedGraphOpsDefinitions& ops_definitions,
169 const IRemoteFusedGraphOpsDefinitions& ops_definitions,
Dgraph_transferer.cc92 const IRemoteFusedGraphOpsDefinitions& ops_definitions, in LoadGraphFromProto() argument
141 ops_definitions, shape_refiner, *node, false, input_node_info_list, in LoadGraphFromProto()
199 const IRemoteFusedGraphOpsDefinitions& ops_definitions, in LoadGraphFromProtoFile() argument
235 return LoadGraphFromProto(ops_definitions, graph_def, input_node_info_list, in LoadGraphFromProtoFile()
422 const IRemoteFusedGraphOpsDefinitions& ops_definitions, in RegisterNode() argument
432 RegisterInputNode(ops_definitions, shape_refiner, node); in RegisterNode()
437 RegisterPadNode(ops_definitions, shape_refiner, node); in RegisterNode()
439 RegisterNodeWithPaddingAndStrides(ops_definitions, shape_refiner, node); in RegisterNode()
441 RegisterNodeWithRank(ops_definitions, shape_refiner, node); in RegisterNode()
443 RegisterFlattenNode(ops_definitions, shape_refiner, node); in RegisterNode()
[all …]
Dhexagon_graph_execution_test.cc409 const IRemoteFusedGraphOpsDefinitions* ops_definitions = in TEST() local
421 *ops_definitions, MODEL_FILENAME, inputs, output_node_names, in TEST()
442 const IRemoteFusedGraphOpsDefinitions* ops_definitions = in TEST() local
454 *ops_definitions, MODEL_WITH_QUANTIZED_INPUT_FILENAME, inputs, in TEST()
476 const IRemoteFusedGraphOpsDefinitions* ops_definitions = in TEST() local
488 *ops_definitions, MODEL_FILENAME, inputs, output_node_names, in TEST()
507 const IRemoteFusedGraphOpsDefinitions* ops_definitions = in TEST() local
545 const IRemoteFusedGraphOpsDefinitions* ops_definitions = in TEST() local
556 *ops_definitions, MODEL_FILENAME, inputs, output_node_names, in TEST()
575 *ops_definitions, MODEL_FILENAME, inputs, output_node_names, in TEST()
Dgraph_transferer_test.cc366 const IRemoteFusedGraphOpsDefinitions& ops_definitions = in TEST() local
368 const int total_ops_count = ops_definitions.GetTotalOpsCount(); in TEST()
373 const IRemoteFusedGraphOpsDefinitions* ops_definitions = in TEST() local
393 *ops_definitions, filename, input_node_info_list, output_node_names, in TEST()
436 const IRemoteFusedGraphOpsDefinitions* ops_definitions = in TEST() local
459 *ops_definitions, filename, input_node_info_list, output_node_names, in TEST()
471 *ops_definitions, filename, input_node_info_list, output_node_names, in TEST()
Dgraph_transfer_utils.h42 const IRemoteFusedGraphOpsDefinitions& ops_definitions,
Dgraph_transfer_utils.cc101 const IRemoteFusedGraphOpsDefinitions& ops_definitions, in BuildFusedGraphDef() argument
/external/tensorflow/tensorflow/core/kernels/
Dremote_fused_graph_execute_utils.h296 const IRemoteFusedGraphOpsDefinitions& ops_definitions);
Dremote_fused_graph_execute_utils.cc1414 const IRemoteFusedGraphOpsDefinitions& ops_definitions) { in BuildNodeMapFromOpsDefinitions() argument
1424 if (ops_definitions.GetOpIdFor( in BuildNodeMapFromOpsDefinitions()