Home
last modified time | relevance | path

Searched defs:all_ops (Results 1 – 12 of 12) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/experimental/tac/transforms/
H A Draise_target_subgraphs.cc63 llvm::SetVector<Operation*> all_ops; member
136 void CollectInputs(const llvm::SetVector<Operation*>& all_ops, in CollectInputs()
163 void CollectOutputs(const llvm::SetVector<Operation*>& all_ops, in CollectOutputs()
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/
H A Dmerge_control_flow.cc80 llvm::SmallSetVector<Operation*, 4> all_ops; in GetAllOpsFromIf() local
207 auto all_ops = GetAllOpsFromIf(*(std::next(iter))); in CanAddToIfSegment() local
/aosp_15_r20/external/angle/third_party/spirv-tools/src/test/reduce/
H A Doperand_to_constant_test.cpp278 const auto all_ops = in TEST() local
/aosp_15_r20/external/deqp-deps/SPIRV-Tools/test/reduce/
Doperand_to_constant_test.cpp278 const auto all_ops = in TEST() local
/aosp_15_r20/external/swiftshader/third_party/SPIRV-Tools/test/reduce/
H A Doperand_to_constant_test.cpp278 const auto all_ops = in TEST() local
/aosp_15_r20/external/pytorch/tools/pyi/
H A Dgen_pyi.py222 all_ops = binary_ops + comparison_ops + unary_ops + to_py_type_ops variable
/aosp_15_r20/external/ComputeLibrary/src/dynamic_fusion/sketch/utils/
H A DDependencyGraph.h436 std::vector<OperatorId> all_ops() const in all_ops() function
/aosp_15_r20/external/tensorflow/tensorflow/compiler/mlir/lite/quantization/
H A Dquantization_context.cc63 std::vector<quantfork::QuantizeRegionOp> all_ops; in GetAllOps() local
/aosp_15_r20/external/pytorch/test/inductor/
H A Dtest_foreach.py58 all_ops = parametrize( variable
/aosp_15_r20/external/tensorflow/tensorflow/compiler/jit/
H A Dmark_for_compilation_pass.cc1273 absl::flat_hash_set<string> all_ops(vall_ops.begin(), vall_ops.end()); in FindCompilationCandidates() local
H A Dmark_for_compilation_pass_test.cc1945 absl::flat_hash_set<string> all_ops(vall_ops.begin(), vall_ops.end()); in TEST() local
/aosp_15_r20/external/pytorch/aten/src/ATen/core/op_registration/
H A Dop_registration_test.cpp2144 auto all_ops = Dispatcher::singleton().getRegistrationsForDispatchKey(std::nullopt); in TEST() local