Home
last modified time | relevance | path

Searched defs:all_inputs (Results 1 – 15 of 15) sorted by relevance

/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/model_tracer/
H A DMobileModelRunner.cpp15 c10::List<at::IValue> all_inputs = bundled_inputs.toList(); in ivalue_to_bundled_inputs() local
45 c10::Dict<at::IValue, at::IValue> all_inputs = bundled_inputs.toGenericDict(); in ivalue_to_bundled_inputs_map() local
/aosp_15_r20/external/pytorch/torch/csrc/jit/codegen/fuser/
H A Dexecutor.cpp191 const at::ArrayRef<IValue>& all_inputs, in launchFusion()
332 auto all_inputs = last(stack, spec.nInputs()); in runFusion() local
/aosp_15_r20/external/skia/gn/
H A Dminify_sksl_tests.py25 all_inputs = shlex.split(reader.read()) variable
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/
H A Dpack.cc100 VectorOfTensors<T> all_inputs(*context, *node->inputs); in PackImpl() local
H A Dadd_n.cc106 VectorOfTensors<T> all_inputs(*context, *node->inputs); in EvalAddN() local
H A Dconcatenation.cc175 VectorOfTensors<int8_t> all_inputs(*context, *node->inputs); in Prepare() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/gl/compiler/
H A Dfuse_auto_input.cc106 absl::flat_hash_set<ValueId> all_inputs; in ApplyToNode() local
/aosp_15_r20/external/pytorch/binaries/
H A Dspeed_benchmark_torch.cc269 auto all_inputs = (*get_method)({}).toList(); in main() local
/aosp_15_r20/external/pytorch/benchmarks/static_runtime/
H A Dtest_cpu_fusion.cc104 std::vector<std::vector<std::pair<int, int>>> all_inputs; in TEST() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/c/
H A Dc_test.c189 float* all_inputs[] = {inputs_1, inputs_2, inputs_3}; in TestRepeatResizeInputTensor() local
/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/
H A Dsummarize_graph_main.cc67 std::vector<const NodeDef*> all_inputs(placeholders); in PrintBenchmarkUsage() local
/aosp_15_r20/external/mesa3d/src/intel/compiler/
H A Dintel_clc.c585 char *all_inputs = NULL; in main() local
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/
H A Dsegment.cc472 const Node* node, const std::vector<OpInfo::TensorProperties>& all_inputs) { in GetInputsToDeterminateBatchSize()
/aosp_15_r20/external/pytorch/torch/autograd/
H A Dgradcheck.py987 func, func_out, all_inputs, inputs_indices, all_u, eps, nondet_tol argument
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/
H A Dtraining_v1.py2542 def _compile_from_inputs(self, all_inputs, target, orig_inputs, orig_target): argument