Home
last modified time | relevance | path

Searched refs:ninputs (Results 1 – 15 of 15) sorted by relevance

/external/tensorflow/tensorflow/java/src/main/native/
Dgraph_jni.cc204 const TF_Output* const outputs, const int ninputs, in buildSubgraph() argument
213 jlongArray input_handles = env->NewLongArray(ninputs); in buildSubgraph()
214 jintArray input_indices = env->NewIntArray(ninputs); in buildSubgraph()
226 for (int i = 0; i < ninputs; ++i) { in buildSubgraph()
264 int ninputs = env->GetArrayLength(input_handles); in Java_org_tensorflow_Graph_whileLoop() local
266 std::unique_ptr<TF_Output[]> inputs(new TF_Output[ninputs]); in Java_org_tensorflow_Graph_whileLoop()
268 ninputs); in Java_org_tensorflow_Graph_whileLoop()
272 TF_WhileParams params = TF_NewWhile(g, inputs.get(), ninputs, status); in Java_org_tensorflow_Graph_whileLoop()
278 params.cond_inputs, &params.cond_output, params.ninputs, 1); in Java_org_tensorflow_Graph_whileLoop()
283 params.body_outputs, params.ninputs, params.ninputs); in Java_org_tensorflow_Graph_whileLoop()
[all …]
Dsession_jni.cc141 const jint ninputs = env->GetArrayLength(input_tensor_handles); in Java_org_tensorflow_Session_run() local
145 std::unique_ptr<TF_Output[]> inputs(new TF_Output[ninputs]); in Java_org_tensorflow_Session_run()
146 std::unique_ptr<TF_Tensor* []> input_values(new TF_Tensor*[ninputs]); in Java_org_tensorflow_Session_run()
154 ninputs); in Java_org_tensorflow_Session_run()
156 ninputs); in Java_org_tensorflow_Session_run()
177 static_cast<int>(ninputs), outputs.get(), output_values.get(), in Java_org_tensorflow_Session_run()
/external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/
DGraph.java280 int ninputs = inputHandles.length; in buildSubgraph() local
282 Output<?>[] inputs = new Output<?>[ninputs]; in buildSubgraph()
289 for (int i = 0; i < ninputs; i++) { in buildSubgraph()
324 int ninputs = inputs.length; in whileLoop() local
325 long[] inputHandles = new long[ninputs]; in whileLoop()
326 int[] inputIndices = new int[ninputs]; in whileLoop()
327 Output<?>[] outputs = new Output<?>[ninputs]; in whileLoop()
332 for (int i = 0; i < ninputs; i++) { in whileLoop()
340 for (int i = 0, j = ninputs; i < ninputs; ++i, ++j) { in whileLoop()
/external/tensorflow/tensorflow/c/
Dc_api_function.cc51 const TF_Graph* fn_body, const char* fn_name, int ninputs, in ProcessInputs() argument
55 input_tensors->reserve(ninputs); in ProcessInputs()
56 for (int i = 0; i < ninputs; ++i) { in ProcessInputs()
153 const TF_Operation* const* opers, int ninputs, const TF_Output* inputs, in TF_GraphToFunctionWithControlOutputs() argument
163 status->status = tensorflow::ProcessInputs(fn_body, fn_name, ninputs, inputs, in TF_GraphToFunctionWithControlOutputs()
228 int ninputs, const TF_Output* inputs, in TF_GraphToFunction() argument
234 fn_body, fn_name, append_hash_to_fn_name, num_opers, opers, ninputs, in TF_GraphToFunction()
Dc_api.h912 const int ninputs; member
952 int ninputs,
1109 const TF_Operation* const* opers, int ninputs, const TF_Output* inputs,
1129 const TF_Operation* const* opers, int ninputs, const TF_Output* inputs,
1277 const TF_Output* inputs, TF_Tensor* const* input_values, int ninputs,
1299 const TF_Output* inputs, int ninputs,
1314 const TF_Output* inputs, TF_Tensor* const* input_values, int ninputs,
1354 int ninputs, const char** output_names,
1361 const char** input_names, int ninputs,
1370 int ninputs, const char** output_names,
Dwhile_loop_test.cc35 void Init(int ninputs) { in Init() argument
37 DCHECK_GT(ninputs, 0); in Init()
39 for (int i = 0; i < ninputs; ++i) { in Init()
57 outputs_.resize(ninputs, {nullptr, -1}); in Init()
144 EXPECT_EQ(params_->ninputs, 2); in TEST_F()
183 for (int i = 0; i < params_->ninputs; ++i) { in TEST_F()
Dc_api.cc382 const int ninputs = input_pairs->size(); in TF_Run_Inputs() local
383 for (int i = 0; i < ninputs; ++i) { in TF_Run_Inputs()
470 const char** c_input_names, TF_Tensor** c_inputs, int ninputs, in TF_Run() argument
477 std::vector<std::pair<string, Tensor>> input_pairs(ninputs); in TF_Run()
479 for (int i = 0; i < ninputs; ++i) { in TF_Run()
496 const char** c_input_names, int ninputs, in TF_PRunSetup() argument
504 std::vector<string> input_names(ninputs); in TF_PRunSetup()
507 for (int i = 0; i < ninputs; ++i) { in TF_PRunSetup()
528 const char** c_input_names, TF_Tensor** c_inputs, int ninputs, in TF_PRun() argument
535 std::vector<std::pair<string, Tensor>> input_pairs(ninputs); in TF_PRun()
[all …]
Dc_api_test.cc1646 TF_Output* inputs, int ninputs, TF_Output* outputs, in AddGradients() argument
1655 ninputs, grad_inputs, s_, grad_outputs); in AddGradients()
1658 ninputs, nullptr, s_, grad_outputs); in AddGradients()
/external/pdfium/core/fpdfapi/page/
Dcpdf_function.cpp126 uint32_t ninputs, in Call() argument
129 if (m_nInputs != ninputs) in Call()
Dcpdf_function.h35 uint32_t ninputs,
/external/tensorflow/tensorflow/python/ops/
Daccumulate_n_benchmark.py120 def _RunBenchmark(self, tag, input_fn, sizes, ninputs, repeats): argument
122 for ninput in ninputs:
/external/mesa3d/src/gallium/drivers/r600/sfn/
Dsfn_nir.h56 unsigned ninputs() const;
Dsfn_shader_fragment.cpp249 size_t ninputs = m_shaderio.inputs().size(); in do_allocate_reserved_registers() local
250 for (size_t k = 0; k < ninputs; ++k) { in do_allocate_reserved_registers()
/external/mesa3d/src/freedreno/ir3/
Dir3_context.h63 unsigned ninputs, noutputs; member
Dir3_compiler_nir.c2981 compile_assert(ctx, idx < ctx->ninputs); in setup_input()
3269 ctx->ninputs = ctx->s->num_inputs * 4; in emit_instructions()
3271 ctx->inputs = rzalloc_array(ctx, struct ir3_instruction *, ctx->ninputs); in emit_instructions()
3603 for (int i = 0; i < ctx->ninputs; i++) { in ir3_compile_shader_nir()
3683 for (unsigned i = 0; i < ctx->ninputs; i++) { in ir3_compile_shader_nir()
3696 ret = ir3_ra(so, ctx->inputs, ctx->ninputs); in ir3_compile_shader_nir()