| /external/tensorflow/tensorflow/tools/graph_transforms/ |
| 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()
|
| 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()
|
| 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()
|
| 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()
|
| D | fold_batch_norms.cc | 50 std::vector<NodeDef>* new_nodes) { in FoldBatchNorms()
117 {}, &replaced_graph_def)); in FoldBatchNorms()
|
| D | round_weights.cc | 42 std::vector<NodeDef>* new_nodes) { in RoundWeights()
108 {}, output_graph_def)); in RoundWeights()
|
| D | remove_nodes.cc | 72 std::vector<NodeDef>* new_nodes) { in RemoveNodes()
99 {true}, &replaced_graph_def)); in RemoveNodes()
|
| D | quantize_weights.cc | 45 std::vector<NodeDef>* new_nodes) { in QuantizeWeights()
138 {}, output_graph_def)); in QuantizeWeights()
|
| D | flatten_atrous.cc | 52 std::vector<NodeDef>* new_nodes) { in FlattenAtrousConv()
132 {}, &replaced_graph_def)); in FlattenAtrousConv()
|
| D | strip_unused_nodes.cc | 120 std::set<string> input_nodes; in StripUnusedNodes() local
|
| D | sparsify_gather.cc | 260 std::vector<NodeDef>* new_nodes) { in SparsifyGatherInternal()
449 {true}, &replaced_graph_def)); in SparsifyGatherInternal()
|
| D | fold_constants_lib.cc | 54 StringPieceSet input_nodes; in ReplaceSendRecvs() local
|
| /external/tensorflow/tensorflow/c/ |
| 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
|
| /external/libaom/av1/encoder/ |
| D | ml.c | 30 void av1_nn_predict_c(const float *input_nodes, in av1_nn_predict_c()
122 const float *input_nodes = feature; in av1_nn_predict_v2() local
|
| /external/tensorflow/tensorflow/core/tfrt/utils/ |
| 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
|
| /external/tensorflow/tensorflow/core/tfrt/graph_executor/ |
| D | graph_executor.h | 99 tensorflow::GraphImportConfig::InputArrays input_nodes; member
|
| /external/tensorflow/tensorflow/compiler/mlir/lite/quantization/ |
| D | quantization_config.cc | 95 std::vector<std::string> input_nodes = absl::StrSplit(node_names, ','); in ParseInputNodeQuantSpecs() local
|
| /external/tensorflow/tensorflow/core/grappler/optimizers/ |
| 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
|
| D | auto_parallel.cc | 163 std::vector<const NodeDef*> input_nodes; in Initialize() local
|
| /external/tensorflow/tensorflow/core/kernels/data/ |
| D | window_dataset.cc | 94 std::vector<Node*> input_nodes; in AsGraphDefInternal() local
|
| /external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| D | configure_tpu_embedding_rewrite_pass.cc | 82 absl::Span<Node* const> input_nodes, in AddSetupPropagationEmbeddingNode()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | cluster_function_library_runtime.cc | 57 std::vector<Node*> input_nodes; in ConstructFunctionGraph() local
|
| /external/tensorflow/tensorflow/core/tfrt/saved_model/ |
| D | saved_model.cc | 721 tensorflow::GraphImportConfig::InputArrays input_nodes; member
852 const tensorflow::GraphImportConfig::InputArrays& input_nodes, in ImportSubgraph()
|
| /external/libaom/av1/encoder/x86/ |
| D | ml_sse3.c | 153 void av1_nn_predict_sse3(const float *input_nodes, in av1_nn_predict_sse3()
|
| /external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| D | choose_fastest_dataset_op.cc | 180 std::vector<Node*> input_nodes; in AsGraphDefInternal() local
|