Home
last modified time | relevance | path

Searched refs:noutputs (Results 1 – 9 of 9) sorted by relevance

/external/tensorflow/tensorflow/c/
Dc_api_function.cc88 int noutputs, const TF_Output* outputs, in ProcessOutputs() argument
91 output_tensors->reserve(noutputs); in ProcessOutputs()
92 for (int i = 0; i < noutputs; ++i) { in ProcessOutputs()
154 int noutputs, const TF_Output* outputs, const char* const* output_names, in TF_GraphToFunctionWithControlOutputs() argument
169 status->status = tensorflow::ProcessOutputs(fn_body, fn_name, noutputs, in TF_GraphToFunctionWithControlOutputs()
176 output_names_vec.reserve(noutputs); in TF_GraphToFunctionWithControlOutputs()
177 for (int i = 0; i < noutputs; ++i) { in TF_GraphToFunctionWithControlOutputs()
223 int noutputs, const TF_Output* outputs, in TF_GraphToFunction() argument
229 inputs, noutputs, outputs, output_names, 0, nullptr, nullptr, opts, in TF_GraphToFunction()
Dc_api.h1101 int noutputs, const TF_Output* outputs, const char* const* output_names,
1121 int noutputs, const TF_Output* outputs, const char* const* output_names,
1270 const TF_Output* outputs, TF_Tensor** output_values, int noutputs,
1292 const TF_Output* outputs, int noutputs,
1307 const TF_Output* outputs, TF_Tensor** output_values, int noutputs,
1346 TF_Tensor** outputs, int noutputs,
1353 const char** output_names, int noutputs,
1362 TF_Tensor** outputs, int noutputs,
Dc_api.cc371 static void TF_Run_Setup(int noutputs, TF_Tensor** c_outputs, in TF_Run_Setup() argument
374 for (int i = 0; i < noutputs; ++i) { in TF_Run_Setup()
418 const int noutputs = output_tensor_names.size(); in TF_Run_Helper() local
419 std::vector<Tensor> outputs(noutputs); in TF_Run_Helper()
454 for (int i = 0; i < noutputs; ++i) { in TF_Run_Helper()
472 const char** c_output_names, TF_Tensor** c_outputs, int noutputs, in TF_Run() argument
476 TF_Run_Setup(noutputs, c_outputs, status); in TF_Run()
482 std::vector<string> output_names(noutputs); in TF_Run()
483 for (int i = 0; i < noutputs; ++i) { in TF_Run()
498 const char** c_output_names, int noutputs, in TF_PRunSetup() argument
[all …]
Dc_api_test.cc1610 int noutputs, TF_Output* grad_outputs) { in AddGradients() argument
1617 TF_AddGradientsWithPrefix(graph_, prefix, outputs, noutputs, inputs, in AddGradients()
1620 TF_AddGradientsWithPrefix(graph_, prefix, outputs, noutputs, inputs, in AddGradients()
/external/tensorflow/tensorflow/java/src/main/native/
Dsession_jni.cc142 const jint noutputs = env->GetArrayLength(output_tensor_handles); in Java_org_tensorflow_Session_run() local
147 std::unique_ptr<TF_Output[]> outputs(new TF_Output[noutputs]); in Java_org_tensorflow_Session_run()
148 std::unique_ptr<TF_Tensor* []> output_values(new TF_Tensor*[noutputs]); in Java_org_tensorflow_Session_run()
158 outputs.get(), noutputs); in Java_org_tensorflow_Session_run()
178 static_cast<int>(noutputs), targets.get(), in Java_org_tensorflow_Session_run()
190 for (int i = 0; i < noutputs; ++i) { in Java_org_tensorflow_Session_run()
Dgraph_jni.cc205 const int noutputs) { in buildSubgraph() argument
215 jlongArray output_handles = env->NewLongArray(noutputs); in buildSubgraph()
216 jintArray output_indices = env->NewIntArray(noutputs); in buildSubgraph()
231 for (int i = 0; i < noutputs; ++i) { in buildSubgraph()
/external/tensorflow/tensorflow/java/src/main/java/org/tensorflow/
DGraph.java281 int noutputs = outputHandles.length; in buildSubgraph() local
283 Output<?>[] outputs = new Output<?>[noutputs]; in buildSubgraph()
284 long[] outputHandlesAndIndices = new long[noutputs * 2]; in buildSubgraph()
294 for (int i = 0; i < noutputs; i++) { in buildSubgraph()
301 for (int i = 0, j = noutputs; i < noutputs; i++, j++) { in buildSubgraph()
/external/mesa3d/src/freedreno/ir3/
Dir3_context.h63 unsigned ninputs, noutputs; member
Dir3_compiler_nir.c3209 compile_assert(ctx, idx < ctx->noutputs); in setup_output()
3270 ctx->noutputs = ctx->s->num_outputs * 4; in emit_instructions()
3272 ctx->outputs = rzalloc_array(ctx, struct ir3_instruction *, ctx->noutputs); in emit_instructions()
3521 assert((ctx->noutputs % 4) == 0); in ir3_compile_shader_nir()
3526 for (unsigned i = 0; i < ctx->noutputs; i += 4) { in ir3_compile_shader_nir()