Home
last modified time | relevance | path

Searched defs:output_node (Results 26 – 45 of 45) sorted by relevance

12

/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/
H A Dremapper.cc2602 auto* output_node = in AddFusedMatMulBiasAddAndGelu() local
2649 auto* output_node = in AddMklLayerNorm() local
3144 auto* output_node = in AddFusedBatchMatMul() local
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_max_autotune.py250 def output_node(self) -> "TensorBox": # noqa: F821 member in TestMaxAutotune.test_precompilation_threads.FakeChoiceCaller
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/
H A Dpeephole.cpp776 auto output_node = output->node(); in eraseTupleConstruct() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/
H A Dfrozen_ops_to_mkldnn.cpp1087 for (auto output_node : output_nodes) { in scanNode() local
/aosp_15_r20/external/crosvm/devices/src/virtio/
H A Diommu.rs122 output_node: u16, field
/aosp_15_r20/external/tensorflow/tensorflow/core/grappler/costs/
H A Dvirtual_scheduler.cc854 for (auto* output_node : port_num_output_pair.second) { in GetOutputNodes() local
H A Dgraph_properties.cc931 NodeDef* output_node = output_nodes[output_arg.node_name]; in UpdateFunction() local
/aosp_15_r20/external/tensorflow/tensorflow/core/tfrt/utils/
H A Dtfrt_graph_execution_state_test.cc764 NodeDef* output_node = nullptr; member
/aosp_15_r20/external/tensorflow/tensorflow/core/data/
H A Dhash_utils_test.cc164 NodeDef* output_node = g.add_node(); in TEST_F() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/passes/quantization/
H A Dinsert_observers.cpp1002 Node* output_node = reset_observer_graph->createNone(); in insertObserverResetMinMax() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/python/
H A Dpython_sugared_value.cpp799 for (auto& output_node : output_nodes) { in call() local
/aosp_15_r20/external/pytorch/torch/_inductor/
H A Dutils.py768 def output_node(gm: torch.fx.GraphModule): function
H A Dpattern_matcher.py215 def output_node(self) -> torch.fx.Node: member in Match
H A Dir.py3636 def output_node(self) -> "TensorBox": member in ChoiceCaller
/aosp_15_r20/external/tensorflow/tensorflow/core/framework/
H A Dfunction.cc479 void AddInput(int node_index, int output_node, int output_index) { in AddInput()
/aosp_15_r20/external/tensorflow/tensorflow/core/tpu/graph_rewrite/
H A Ddistributed_tpu_rewrite_pass.cc2357 Node* output_node = partitioned_output_nodes[i]; in AssignArgsAndRetvalsToCores() local
/aosp_15_r20/external/bcc/libbpf-tools/x86/
H A Dvmlinux.h88936 u16 output_node; member
88944 u16 output_node; member
H A Dvmlinux_518.h88936 u16 output_node; member
88944 u16 output_node; member
/aosp_15_r20/external/bcc/libbpf-tools/arm64/
H A Dvmlinux_608.h133880 u16 output_node; member
133888 u16 output_node; member
H A Dvmlinux.h133880 u16 output_node; member
133888 u16 output_node; member

12