Searched defs:op_node (Results 1 – 10 of 10) sorted by relevance
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/ |
| H A D | variadic_ops.cpp | 55 void recordSchema(Node* op_node) { in recordSchema()
83 bool allListInputsAreValid(Node* op_node) { in allListInputsAreValid()
120 void deleteOpNodeAndLists(Node* op_node) { in deleteOpNodeAndLists()
140 bool replaceWithVariadicOp(Node* op_node) { in replaceWithVariadicOp()
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/transformations/ |
| H A D | merge_padding_with.cc | 56 Node* op_node = sequence.back(); in ApplyToNodesSequence() local
|
| H A D | remove_noop.cc | 54 Node* op_node = sequence.back(); in ApplyToNodesSequence() local
|
| /aosp_15_r20/external/executorch/backends/qualcomm/builders/ |
| H A D | node_visitor.py | 92 def get_tensor(self, input_node, op_node, idx=None): argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/ |
| H A D | op_builder.h | 295 const OpNode* op_node = builders_[i]->Build(); in Build() local
|
| /aosp_15_r20/external/mesa3d/src/amd/compiler/ |
| H A D | aco_lower_to_cssa.cpp | 337 merge_node& op_node = ctx.merge_node_table[copy.op.tempId()]; in try_coalesce_copy() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/ops/ |
| H A D | math_grad_test.cc | 43 Status Unary(const FDH::Node& op_node, const Tensor& x, const DataType dst, in Unary()
98 const FDH::Node op_node = {{"y"}, op, {"x"}, {{"T", x.dtype()}}}; in Unary() local
111 const FDH::Node op_node = { in SymCastGrad() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/jit/passes/onnx/ |
| H A D | unpack_quantized_weights.cpp | 726 auto op_node = match_vmap.at(vmap.at("r"))->node(); in insertPermutesHelper() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | loop_optimizer.cc | 554 Status EvaluateBoolOpForConstantOperands(const NodeDef& op_node, in EvaluateBoolOpForConstantOperands()
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/ |
| H A D | dtensor_device_util.cc | 794 tensorflow::Node* op_node = graph->AddNode(op_node_def, &status); in PrepareGraphForMlir() local
|