| /aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/ |
| H A D | backports.cc | 36 std::vector<NodeDef>* new_nodes) { in BackportConcatV2Transform()
71 const std::set<string>& input_nodes, in BackportTensorArrayV3Transform()
73 std::vector<NodeDef>* new_nodes) { in BackportTensorArrayV3Transform()
117 [](const NodeMatch& match, const std::set<string>& input_nodes, in BackportTensorArrayV3Transform()
119 std::vector<NodeDef>* new_nodes) { in BackportTensorArrayV3Transform()
126 {true}, output_graph_def)); in BackportTensorArrayV3Transform()
|
| H A D | fuse_convolutions.cc | 47 std::vector<NodeDef>* new_nodes) { in FuseResizePadAndConv()
94 [](const NodeMatch& match, const std::set<string>& input_nodes, in FuseResizeAndConv()
96 std::vector<NodeDef>* new_nodes) { in FuseResizeAndConv()
154 [](const NodeMatch& match, const std::set<string>& input_nodes, in FusePadAndConv()
156 std::vector<NodeDef>* new_nodes) { in FusePadAndConv()
186 {}, &replaced_graph_def)); in FusePadAndConv()
|
| H A D | quantize_nodes.cc | 282 std::vector<NodeDef>* new_nodes) { in RemoveRedundantQuantizations()
302 {true}, &replaced_graph_def)); in RemoveRedundantQuantizations()
405 std::vector<NodeDef>* new_nodes) { in ConvertFakeQuantsToRequantize()
452 {}, output_graph_def)); in ConvertFakeQuantsToRequantize()
499 std::vector<NodeDef>* new_nodes) { in MergeAdjacentRequantizes()
522 {}, output_graph_def)); in MergeAdjacentRequantizes()
549 std::vector<NodeDef>* new_nodes) { in HoistFakeQuants()
578 {}, &hoisted_graph_def)); in HoistFakeQuants()
698 std::vector<NodeDef>* new_nodes) { in QuantizeNodes()
923 {}, &quantized_graph_def)); in QuantizeNodes()
|
| H A D | fold_old_batch_norms.cc | 313 std::vector<NodeDef>* new_nodes) { in FoldOldBatchNorms()
318 {}, &replaced_graph_def)); in FoldOldBatchNorms()
350 std::vector<NodeDef>* new_nodes) { in FoldOldBatchNorms()
355 {}, &replaced_graph_def)); in FoldOldBatchNorms()
393 std::vector<NodeDef>* new_nodes) { in FoldOldBatchNorms()
398 {}, &replaced_graph_def)); in FoldOldBatchNorms()
|
| H A D | fold_batch_norms.cc | 50 std::vector<NodeDef>* new_nodes) { in FoldBatchNorms()
117 {}, &replaced_graph_def)); in FoldBatchNorms()
|
| H A D | remove_nodes.cc | 72 std::vector<NodeDef>* new_nodes) { in RemoveNodes()
99 {true}, &replaced_graph_def)); in RemoveNodes()
|
| H A D | round_weights.cc | 42 std::vector<NodeDef>* new_nodes) { in RoundWeights()
108 {}, output_graph_def)); in RoundWeights()
|
| H A D | quantize_weights.cc | 45 std::vector<NodeDef>* new_nodes) { in QuantizeWeights()
138 {}, output_graph_def)); in QuantizeWeights()
|
| H A D | flatten_atrous.cc | 52 std::vector<NodeDef>* new_nodes) { in FlattenAtrousConv()
132 {}, &replaced_graph_def)); in FlattenAtrousConv()
|
| H A D | sparsify_gather.cc | 260 std::vector<NodeDef>* new_nodes) { in SparsifyGatherInternal()
449 {true}, &replaced_graph_def)); in SparsifyGatherInternal()
|
| H A D | strip_unused_nodes.cc | 120 std::set<string> input_nodes; in StripUnusedNodes() local
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | select_algorithm.py | 122 input_nodes, argument
605 input_nodes, argument
828 input_nodes, argument
843 input_nodes, argument
928 input_nodes, argument
1137 def create_inputs_key(input_nodes) -> str: argument
1174 input_nodes, argument
1400 input_nodes, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/c/ |
| H A D | c_api_function.cc | 54 std::unordered_map<const Node*, std::vector<int>>* input_nodes) in ProcessInputs()
113 const std::unordered_map<const Node*, std::vector<int>>& input_nodes, in ComputeBodyNodes()
163 std::unordered_map<const Node*, std::vector<int>> input_nodes; in TF_GraphToFunctionWithControlOutputs() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/ |
| H A D | graph_partition.cc | 76 absl::flat_hash_map<std::string, NodeInfo>& input_nodes, in PrepareSubgraphForFunctionConversion()
216 const absl::flat_hash_map<std::string, NodeInfo>& input_nodes, in BuildPartitionedCallOp()
458 absl::flat_hash_map<std::string, NodeInfo> input_nodes; in InsertTransferOps() local
|
| /aosp_15_r20/external/libaom/av1/encoder/ |
| H A D | ml.c | 31 void av1_nn_predict_c(const float *input_nodes, in av1_nn_predict_c()
123 const float *input_nodes = feature; in av1_nn_predict_v2() local
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
| H A D | cpp_gemm_template.py | 241 input_nodes, argument
510 input_nodes, argument
|
| H A D | cpp_template.py | 29 input_nodes, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | dependency_optimizer.cc | 196 const NodeDef& node, const std::vector<NodeDef*>& input_nodes, in BypassingNodeIsBeneficial()
385 std::vector<NodeDef*> input_nodes; in OptimizeNode() local
|
| H A D | auto_parallel.cc | 163 std::vector<const NodeDef*> input_nodes; in Initialize() local
|
| /aosp_15_r20/external/pytorch/torch/fx/passes/utils/ |
| H A D | source_matcher_utils.py | 40 input_nodes: List[Node] = field(default_factory=list) variable in SourcePartition
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/saved_model/ |
| H A D | saved_model.cc | 721 tensorflow::GraphImportConfig::InputArrays input_nodes; member
852 const tensorflow::GraphImportConfig::InputArrays& input_nodes, in ImportSubgraph()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/graph_executor/ |
| H A D | graph_executor.h | 99 tensorflow::GraphImportConfig::InputArrays input_nodes; member
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/rocm/ |
| H A D | ck_universal_gemm_template.py | 386 input_nodes, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| H A D | quantization_config.cc | 95 std::vector<std::string> input_nodes = absl::StrSplit(node_names, ','); in ParseInputNodeQuantSpecs() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | subgraph_rewrite.cpp | 14 const std::set<const Node*>& input_nodes, in update_source_range_and_cs_ptr()
|