| /aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/ |
| H A D | vararg_functions.cpp | 108 void format(Stack& stack, size_t num_inputs) { in format()
143 void einsum(Stack& stack, size_t num_inputs) { in einsum()
221 void percentFormat(Stack& stack, size_t num_inputs) { in percentFormat()
274 void tupleConstruct(Stack& stack, size_t num_inputs) { in tupleConstruct()
317 size_t num_inputs) { in namedTupleConstruct()
330 size_t num_inputs) { in listConstruct()
335 [](Stack& stack, const c10::Type& list_type, size_t num_inputs) { in listConstruct()
350 size_t num_inputs) { in dictConstruct()
|
| H A D | register_distributed_ops.cpp | 27 int num_inputs, in prepare_and_call_rpc_op()
248 int num_inputs = node->inputs().size(); in __anon84498b7a0902() local
257 int num_inputs = node->inputs().size(); in __anon84498b7a0b02() local
266 int num_inputs = node->inputs().size(); in __anon84498b7a0d02() local
|
| H A D | register_prim_ops.cpp | 480 size_t num_inputs = pop(stack).toInt(); in __anonbfe5918f1f02() local
823 auto num_inputs = pop(stack).toInt(); in __anonbfe5918f3502() local
846 auto num_inputs = pop(stack).toInt(); in __anonbfe5918f3602() local
858 auto num_inputs = pop(stack).toInt(); in __anonbfe5918f3702() local
2429 size_t num_inputs = pop(stack).toInt(); in __anonbfe5918f7802() local
2672 auto num_inputs = pop(stack).toInt(); in __anonbfe5918f9202() local
2721 auto num_inputs = pop(stack).toInt(); in __anonbfe5918f9602() local
2749 auto num_inputs = pop(stack).toInt(); in __anonbfe5918f9702() local
2765 auto num_inputs = pop(stack).toInt(); in __anonbfe5918f9802() local
|
| /aosp_15_r20/external/libaom/av1/encoder/arm/ |
| H A D | ml_neon.c | 32 static void nn_propagate_8to1(int num_inputs, const float *const inputs, in nn_propagate_8to1()
62 static void nn_propagate_xto1(int num_inputs, const float *const inputs, in nn_propagate_xto1()
97 static void nn_propagate_xsto1(int num_inputs, const float *const inputs, in nn_propagate_xsto1()
116 static void nn_propagate_4to1(int num_inputs, const float *const inputs, in nn_propagate_4to1()
142 static void nn_propagate_4to4(int num_inputs, const float *const inputs, in nn_propagate_4to4()
180 static void nn_propagate_4to8(const int num_inputs, const float *const inputs, in nn_propagate_4to8()
225 static void nn_propagate_8to4(const int num_inputs, const float *const inputs, in nn_propagate_8to4()
272 int num_inputs = nn_config->num_inputs; in av1_nn_predict_neon() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/data/experimental/ |
| H A D | choose_fastest_dataset_op.cc | 55 for (size_t i = 1, num_inputs = inputs.size(); i < num_inputs; ++i) { in MakeDataset() local
68 for (size_t i = 1, num_inputs = inputs.size(); i < num_inputs; ++i) { in MakeDataset() local
96 for (size_t i = 1, num_inputs = inputs.size(); i < num_inputs; ++i) { in MakeDataset() local
205 for (size_t i = 0, num_inputs = dataset()->inputs_.size(); in Initialize() local
318 for (size_t i = 0, num_inputs = dataset()->inputs_.size(); in StartThreads() local
350 for (size_t i = 1, num_inputs = histograms_.size(); i < num_inputs; in SelectFastestInputIndex() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/mkl/ |
| H A D | mkl_aggregate_ops.cc | 50 const int num_inputs = ctx->num_inputs() / 2; in CheckInputShape() local
72 const int num_inputs = ctx->num_inputs() / 2; in FindMKLInputIndex() local
87 const int num_inputs = ctx->num_inputs() / 2; in ComputeScalar() local
113 const int num_inputs = ctx->num_inputs() / 2; in Compute() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tests/ |
| H A D | lstm.py | 38 def LSTMCellWeightsShape(num_inputs, num_nodes): argument
122 def RandomInputs(batch_size, seq_length, num_inputs): argument
135 def BuildLSTMLayer(batch_size, seq_length, num_inputs, num_nodes): argument
|
| /aosp_15_r20/external/libaom/av1/encoder/x86/ |
| H A D | ml_sse3.c | 63 __m128 *const outputs, const int num_inputs) { in av1_nn_propagate_4to4_sse3()
85 __m128 *const out_l, const int num_inputs) { in av1_nn_propagate_4to8_sse3()
112 const int num_inputs) { in nn_propagate_8to4()
158 int num_inputs = nn_config->num_inputs; in av1_nn_predict_sse3() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/tf2xla/ |
| H A D | const_analysis.cc | 68 int num_inputs = branch_bodies[0]->fdef.signature().input_arg_size(); in CondConstInputIndices() local
102 int num_inputs = fbody->fdef.signature().input_arg_size(); in GetCompileTimeConstInputs() local
154 int num_inputs = fbody->fdef.signature().input_arg_size(); in GetCompileTimeConstInputs() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/ |
| H A D | utils.cc | 177 const int num_inputs = node.input_size(); in HasControlInputs() local
185 const int num_inputs = node.input_size(); in HasRegularInputs() local
193 int num_inputs = 0; in NumNonControlInputs() local
204 int num_inputs = 0; in NumControlInputs() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/python/ops/ |
| H A D | concat_benchmark.py | 31 def build_graph(device, input_shape, variable, num_inputs, axis, grad): argument
77 def _run_graph(self, device, input_shape, variable, num_inputs, axis, grad, argument
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/reference/ |
| H A D | add_n.h | 28 inline void AddN(const RuntimeShape& input_shape, const size_t num_inputs, in AddN()
43 const RuntimeShape& input_shape, const size_t num_inputs, in AddN()
|
| /aosp_15_r20/external/tensorflow/tensorflow/dtensor/cc/ |
| H A D | dtensor_device.cc | 640 const char* operation_name, const int num_inputs, in MaybeHandleDTensorCustomOps()
668 void DTensorDevice::CopyToMesh(TFE_Context* context, int num_inputs, in CopyToMesh()
806 TFE_TensorHandle* DTensorDevice::Pack(TFE_Context* context, int num_inputs, in Pack()
916 TFE_Context* context, int num_inputs, TFE_TensorHandle** indices, in SparsePack()
1836 int num_inputs = TFE_OpGetFlatInputCount(original_op, status); in Execute() local
2118 TFE_TensorHandle* Pack(TFE_Context* context, int num_inputs, in Pack()
2139 TFE_TensorHandle* SparsePack(TFE_Context* context, int num_inputs, in SparsePack()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/grappler/optimizers/ |
| H A D | dependency_optimizer.cc | 163 const int num_inputs = node.input_size(); in NumEdgesIfBypassed() local
202 const int num_inputs = node.input_size(); in BypassingNodeIsBeneficial() local
384 const int num_inputs = node->input_size(); in OptimizeNode() local
|
| /aosp_15_r20/external/libaom/av1/encoder/ |
| H A D | ml.h | 25 int num_inputs; // Number of input nodes, i.e. features. member
40 const int num_inputs; // Number of input nodes, i.e. features. member
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/ |
| H A D | add_n.cc | 50 int num_inputs = NumInputs(node); in Prepare() local
109 int num_inputs = NumInputs(node); in EvalAddN() local
|
| H A D | concatenation_test.cc | 39 int axis, int num_inputs, in BaseConcatenationOpModel()
56 int num_inputs) in BaseConcatenationOpModel()
492 const int num_inputs = input_data_list.size(); in PersistentConcatenationOpModel() local
|
| /aosp_15_r20/external/executorch/runtime/executor/ |
| H A D | method_meta.cpp | 106 size_t MethodMeta::num_inputs() const { in num_inputs() function in executorch::runtime::MethodMeta
111 auto num_inputs = this->num_inputs(); in input_tag() local
|
| /aosp_15_r20/external/pytorch/torch/csrc/autograd/ |
| H A D | autograd.cpp | 114 size_t num_inputs = inputs.size(); in run_backward() local
149 size_t num_inputs = inputs.size(); in run_backward() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/experimental/microfrontend/ |
| H A D | audio_microfrontend_test.cc | 92 int num_inputs() { return n_input_; } in num_inputs() function in tflite::ops::custom::__anon609fb90c0111::MicroFrontendOpModel
122 const int num_inputs = micro_frontend->num_inputs(); in VerifyGoldens() local
|
| /aosp_15_r20/packages/modules/NeuralNetworks/common/cpu_operations/ |
| D | Concatenation.cpp | 56 int num_inputs = inputShapes.size(); in concatenation() local
76 int num_inputs = inputShapes.size(); in concatenation() local
|
| /aosp_15_r20/external/private-join-and-compute/third_party/abseil-cpp-20230125.2/absl/random/internal/ |
| H A D | nanobenchmark.cc | 549 InputVec UniqueInputs(const FuncInput* inputs, const size_t num_inputs) { in UniqueInputs()
596 InputVec ReplicateInputs(const FuncInput* inputs, const size_t num_inputs, in ReplicateInputs()
781 const size_t num_inputs, Result* results, const Params& p) { in Measure()
|
| /aosp_15_r20/external/openscreen/third_party/abseil/src/absl/random/internal/ |
| H A D | nanobenchmark.cc | 549 InputVec UniqueInputs(const FuncInput* inputs, const size_t num_inputs) { in UniqueInputs()
596 InputVec ReplicateInputs(const FuncInput* inputs, const size_t num_inputs, in ReplicateInputs()
781 const size_t num_inputs, Result* results, const Params& p) { in Measure()
|
| /aosp_15_r20/external/angle/third_party/abseil-cpp/absl/random/internal/ |
| H A D | nanobenchmark.cc | 549 InputVec UniqueInputs(const FuncInput* inputs, const size_t num_inputs) { in UniqueInputs()
596 InputVec ReplicateInputs(const FuncInput* inputs, const size_t num_inputs, in ReplicateInputs()
781 const size_t num_inputs, Result* results, const Params& p) { in Measure()
|
| /aosp_15_r20/external/rust/android-crates-io/crates/grpcio-sys/grpc/third_party/abseil-cpp/absl/random/internal/ |
| D | nanobenchmark.cc | 549 InputVec UniqueInputs(const FuncInput* inputs, const size_t num_inputs) { in UniqueInputs()
596 InputVec ReplicateInputs(const FuncInput* inputs, const size_t num_inputs, in ReplicateInputs()
781 const size_t num_inputs, Result* results, const Params& p) { in Measure()
|