| /aosp_15_r20/external/tensorflow/tensorflow/core/data/ |
| H A D | serialization_utils.cc | 59 const string& output_node, Tensor* result) { in FromGraphDef()
278 tstring output_node, serialized_graph_def; in ReadDatasetInternal() local
404 string output_node; in WriteDatasetInternal() local
461 Node* output_node = nullptr; in AsGraphDef() local
|
| H A D | rewrite_utils.cc | 105 const std::string& output_node = tensor.substr(0, tensor.find(':')); in RemoveFakeSinks() local
176 string output_node; in RewriteDataset() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | generic_layout_optimizer_transposer_test.cc | 872 auto* output_node = context.graph_view->GetNode("output"); in TEST_F() local
965 auto* output_node = context.graph_view->GetNode("output"); in TEST_F() local
998 auto* output_node = context.graph_view->GetNode("output"); in TEST_F() local
1051 auto* output_node = context.graph_view->GetNode("output"); in TEST_F() local
1150 auto* output_node = context.graph_view->GetNode("output"); in TEST_F() local
1336 auto* output_node = context.graph_view->GetNode("output"); in TEST_F() local
1384 auto* output_node = context.graph_view->GetNode("output"); in TEST_F() local
1451 auto* output_node = context.graph_view->GetNode("output"); in TEST_F() local
1499 auto* output_node = context.graph_view->GetNode("output"); in TEST_F() local
1651 auto* output_node = context.graph_view->GetNode("i2"); in TEST_F() local
[all …]
|
| H A D | generic_layout_optimizer_test.cc | 252 auto* output_node = graph_view.GetNode("Output"); in TEST_F() local
572 auto* output_node = graph_view.GetNode("output"); in TEST_F() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/graph/ |
| H A D | benchmark_testlib.h | 168 NodeDef output_node = create_node(/*name=*/absl::StrFormat("out%05d", i)); in CreateFaninFanoutNodeGraph() local
177 NodeDef output_node = in CreateFaninFanoutNodeGraph() local
|
| /aosp_15_r20/external/pytorch/torch/_inductor/ |
| H A D | select_algorithm.py | 123 output_node, argument
892 def output_node(self): member in TritonTemplateCaller
976 def output_node(self): member in ExternKernelCaller
1062 def output_node(self) -> ir.TensorBox: member in DataProcessorChoiceCallerWrapper
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/ |
| H A D | parallel_interleave_dataset_op_test.cc | 654 Node* output_node = nullptr; in TEST_F() local
695 Node* output_node = nullptr; in TEST_F() local
|
| H A D | dataset_ops.cc | 149 string output_node; in Compute() local
|
| /aosp_15_r20/external/executorch/backends/qualcomm/_passes/ |
| H A D | recompose_rms_norm.py | 29 def _get_gamma_node(self, output_node): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/ |
| H A D | convert_graph.cc | 223 auto output_node = edge->dst(); in GetEngineInfo() local
548 Node* output_node = graph->FindNodeId(conn.outside_id); in CreateTRTNode() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/ |
| H A D | graph_partition_test.cc | 149 NodeDef* output_node = nullptr; in TEST_F() local
|
| H A D | graph_partition.cc | 242 for (const auto& output_node : output_nodes) { in BuildPartitionedCallOp() local
|
| H A D | tfrt_graph_execution_state.cc | 270 for (const auto& output_node : output_nodes) { in PlaceInputOutputNodesOnHost() local
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/rocm/ |
| H A D | rocm_kernel.py | 228 def output_node(self) -> TensorBox: member in ROCmTemplateCaller
|
| /aosp_15_r20/external/executorch/backends/apple/mps/operators/ |
| H A D | node_visitor.py | 454 def process_output_node( argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/ |
| H A D | utils.h | 190 auto output_node = nodes_[NodeName(output_name)]; in AddOutput() local
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/cuda/ |
| H A D | cuda_kernel.py | 387 def output_node(self) -> TensorBox: member in CUDATemplateCaller
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tfrt/transforms/ |
| H A D | tf_to_tfrt_data.cc | 289 string output_node; in TFDataGraphDefToTFDataMLIR() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/ |
| H A D | encapsulate_xla_computations_pass.cc | 294 Node* output_node = le->dst(); in BuildXlaLaunchOps() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/distributed_runtime/ |
| H A D | cluster_function_library_runtime.cc | 135 Node* output_node; in ConstructFunctionGraph() local
|
| /aosp_15_r20/external/pytorch/torch/_inductor/codegen/ |
| H A D | cpp_template_kernel.py | 375 def output_node(self) -> ir.TensorBox: member in CppTemplateCaller
|
| /aosp_15_r20/external/tensorflow/tensorflow/cc/framework/ |
| H A D | gradients.cc | 242 for (int output_node : output_nodes) { in GetStopBackpropNodes() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/ |
| H A D | host_training_loop_optimization_util.cc | 97 Node* output_node = output_edge->dst(); in ResourceOnlyUsedForTPUExecuteInLoop() local
|
| /aosp_15_r20/external/XNNPACK/src/ |
| H A D | subgraph.c | 912 struct xnn_node* output_node = subgraph->nodes + subgraph->num_nodes - 1; in xnn_subgraph_rewrite_for_fp16() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/common_runtime/ |
| H A D | function_test.cc | 1050 const string output_node = "Func/b/output/_1"; in TEST_F() local
1125 const string output_node = "Func/b/output/_1"; in TEST_F() local
1213 const string output_node = "Func/b/output/_1"; in TEST_F() local
|