/aosp_15_r20/external/pytorch/torch/csrc/jit/mobile/model_tracer/ |
H A D | MobileModelRunner.cpp | 15 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 D | executor.cpp | 191 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 D | minify_sksl_tests.py | 25 all_inputs = shlex.split(reader.read()) variable
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
H A D | pack.cc | 100 VectorOfTensors<T> all_inputs(*context, *node->inputs); in PackImpl() local
|
H A D | add_n.cc | 106 VectorOfTensors<T> all_inputs(*context, *node->inputs); in EvalAddN() local
|
H A D | concatenation.cc | 175 VectorOfTensors<int8_t> all_inputs(*context, *node->inputs); in Prepare() local
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/gl/compiler/ |
H A D | fuse_auto_input.cc | 106 absl::flat_hash_set<ValueId> all_inputs; in ApplyToNode() local
|
/aosp_15_r20/external/pytorch/binaries/ |
H A D | speed_benchmark_torch.cc | 269 auto all_inputs = (*get_method)({}).toList(); in main() local
|
/aosp_15_r20/external/pytorch/benchmarks/static_runtime/ |
H A D | test_cpu_fusion.cc | 104 std::vector<std::vector<std::pair<int, int>>> all_inputs; in TEST() local
|
/aosp_15_r20/external/tensorflow/tensorflow/lite/c/ |
H A D | c_test.c | 189 float* all_inputs[] = {inputs_1, inputs_2, inputs_3}; in TestRepeatResizeInputTensor() local
|
/aosp_15_r20/external/tensorflow/tensorflow/tools/graph_transforms/ |
H A D | summarize_graph_main.cc | 67 std::vector<const NodeDef*> all_inputs(placeholders); in PrintBenchmarkUsage() local
|
/aosp_15_r20/external/mesa3d/src/intel/compiler/ |
H A D | intel_clc.c | 585 char *all_inputs = NULL; in main() local
|
/aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2tensorrt/segment/ |
H A D | segment.cc | 472 const Node* node, const std::vector<OpInfo::TensorProperties>& all_inputs) { in GetInputsToDeterminateBatchSize()
|
/aosp_15_r20/external/pytorch/torch/autograd/ |
H A D | gradcheck.py | 987 func, func_out, all_inputs, inputs_indices, all_u, eps, nondet_tol argument
|
/aosp_15_r20/external/tensorflow/tensorflow/python/keras/engine/ |
H A D | training_v1.py | 2542 def _compile_from_inputs(self, all_inputs, target, orig_inputs, orig_target): argument
|