Home
last modified time | relevance | path

Searched defs:ninputs (Results 1 – 18 of 18) sorted by relevance

/aosp_15_r20/external/tensorflow/tensorflow/c/
H A Dc_api_function.cc52 const TF_Graph* fn_body, const char* fn_name, int ninputs, in ProcessInputs()
154 const TF_Operation* const* opers, int ninputs, const TF_Output* inputs, in TF_GraphToFunctionWithControlOutputs()
229 int ninputs, const TF_Output* inputs, in TF_GraphToFunction()
H A Dc_api.cc382 const int ninputs = input_pairs->size(); in TF_Run_Inputs() local
470 const char** c_input_names, TF_Tensor** c_inputs, int ninputs, in TF_Run()
496 const char** c_input_names, int ninputs, in TF_PRunSetup()
528 const char** c_input_names, TF_Tensor** c_inputs, int ninputs, in TF_PRun()
1955 TF_WhileParams TF_NewWhile(TF_Graph* g, TF_Output* inputs, int ninputs, in TF_NewWhile()
2299 int ninputs, const TF_Output* outputs, in TF_SessionRun()
2339 int ninputs, const TF_Output* outputs, int noutputs, in TF_SessionPRunSetup()
2381 int ninputs, const TF_Output* outputs, in TF_SessionPRun()
H A Dc_api.h927 const int ninputs; member
H A Dwhile_loop_test.cc35 void Init(int ninputs) { in Init()
H A Dc_api_test.cc1677 TF_Output* inputs, int ninputs, TF_Output* outputs, in AddGradients()
/aosp_15_r20/external/tensorflow/tensorflow/java/src/main/native/
H A Dgraph_jni.cc204 const TF_Output* const outputs, const int ninputs, in buildSubgraph()
264 int ninputs = env->GetArrayLength(input_handles); in Java_org_tensorflow_Graph_whileLoop() local
H A Dsession_jni.cc141 const jint ninputs = env->GetArrayLength(input_tensor_handles); in Java_org_tensorflow_Session_run() local
/aosp_15_r20/external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/
H A DGraph.java280 int ninputs = inputHandles.length; in buildSubgraph() local
324 int ninputs = inputs.length; in whileLoop() local
/aosp_15_r20/external/trusty/lk/include/dev/
Dgpio_keypad.h46 int ninputs; member
/aosp_15_r20/external/executorch/kernels/portable/cpu/
H A Dop_stack.cpp55 const size_t ninputs = tensors.size(); in stack_out() local
H A Dop_cat.cpp56 const size_t ninputs = tensors.size(); in cat_out() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cpu/
H A DCatKernel.cpp29 int64_t ninputs = static_cast<int64_t>(tensors.size()); in cat_serial_kernel_impl() local
H A DSerialStackImpl.h36 int64_t ninputs = tensors.size(); in stack_serial_kernel_impl() local
/aosp_15_r20/external/tensorflow/tensorflow/python/ops/
H A Daccumulate_n_benchmark.py115 def _RunBenchmark(self, tag, input_fn, sizes, ninputs, repeats): argument
/aosp_15_r20/external/mesa3d/src/freedreno/ir3/
H A Dir3_context.h46 unsigned ninputs, noutputs; member
/aosp_15_r20/external/mesa3d/src/gallium/drivers/r600/sfn/
H A Dsfn_shader.h210 size_t ninputs() const { return m_inputs.size(); } in ninputs() function
/aosp_15_r20/external/pytorch/torch/csrc/jit/runtime/
H A Dargument_spec.h329 size_t ninputs; member
/aosp_15_r20/external/pytorch/aten/src/ATen/
H A DTensorIterator.h286 int ninputs() const { in ninputs() function