Home
last modified time | relevance | path

Searched defs:output_op (Results 1 – 7 of 7) sorted by relevance

/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/cutlass_extensions/gemm/kernel/
H A DfpA_intB_gemm.h112 typename EpilogueOutputOp::Params output_op; member
172 typename EpilogueOutputOp::Params output_op; member
/aosp_15_r20/external/tensorflow/tensorflow/lite/toco/graph_transformations/
H A Didentify_hardswish.cc61 const auto* output_op = GetOpWithInput(*model, mul_op->outputs[0]); in Run() local
H A Dgroup_bidirectional_sequence_ops.cc84 bool FindUnidirectionalSequenceOp(const Model& model, const Operator& output_op, in FindUnidirectionalSequenceOp()
/aosp_15_r20/external/federated-compute/fcp/tensorflow/
H A Dtf_session_test.cc53 void CheckOutput(TfSession* sess, const std::string& output_op, in CheckOutput()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
H A Dpick_subgraphs.cc118 Operation* output_op = output.getDefiningOp(); in GetProducerCallOpOrNull() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/hexagon/builders/
H A Dop_builder.cc312 auto* output_op = AddNode(); in AddOutputTensors() local
/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dc_api_test.cc1442 TF_Operation* output_op = in TEST() local