Home
last modified time | relevance | path

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

/external/tensorflow/tensorflow/java/src/main/native/
Dsession_jni.cc170 const jint noutputs = env->GetArrayLength(output_tensor_handles); in Java_org_tensorflow_Session_run() local
175 std::unique_ptr<TF_Output[]> outputs(new TF_Output[noutputs]); in Java_org_tensorflow_Session_run()
176 std::unique_ptr<TF_Tensor* []> output_values(new TF_Tensor*[noutputs]); in Java_org_tensorflow_Session_run()
186 outputs.get(), noutputs); in Java_org_tensorflow_Session_run()
206 static_cast<int>(noutputs), targets.get(), in Java_org_tensorflow_Session_run()
218 for (int i = 0; i < noutputs; ++i) { in Java_org_tensorflow_Session_run()
/external/tensorflow/tensorflow/c/
Dc_api_function.cc425 int noutputs, const TF_Output* outputs, in ProcessOutputs() argument
428 output_tensors->reserve(noutputs); in ProcessOutputs()
429 for (int i = 0; i < noutputs; ++i) { in ProcessOutputs()
491 int noutputs, const TF_Output* outputs, in TF_GraphToFunction() argument
506 status->status = tensorflow::ProcessOutputs(fn_body, fn_name, noutputs, in TF_GraphToFunction()
513 output_names_vec.reserve(noutputs); in TF_GraphToFunction()
514 for (int i = 0; i < noutputs; ++i) { in TF_GraphToFunction()
Dc_api.cc713 static void TF_Run_Setup(int noutputs, TF_Tensor** c_outputs, in TF_Run_Setup() argument
716 for (int i = 0; i < noutputs; ++i) { in TF_Run_Setup()
741 const int noutputs = output_tensor_names.size(); in TF_Run_Helper() local
742 std::vector<Tensor> outputs(noutputs); in TF_Run_Helper()
777 for (int i = 0; i < noutputs; ++i) { in TF_Run_Helper()
795 const char** c_output_names, TF_Tensor** c_outputs, int noutputs, in TF_Run() argument
799 TF_Run_Setup(noutputs, c_outputs, status); in TF_Run()
805 std::vector<string> output_names(noutputs); in TF_Run()
806 for (int i = 0; i < noutputs; ++i) { in TF_Run()
821 const char** c_output_names, int noutputs, in TF_PRunSetup() argument
[all …]
Dc_api.h1231 int noutputs, const TF_Output* outputs, const char* const* output_names,
1364 const TF_Output* outputs, TF_Tensor** output_values, int noutputs,
1386 const TF_Output* outputs, int noutputs,
1401 const TF_Output* outputs, TF_Tensor** output_values, int noutputs,
1440 TF_Tensor** outputs, int noutputs,
1447 const char** output_names, int noutputs,
1456 TF_Tensor** outputs, int noutputs,
Dc_api_test.cc1528 TF_Output* outputs, int noutputs, TF_Output* grad_outputs) { in AddGradients() argument
1535 TF_AddGradients(graph_, outputs, noutputs, inputs, ninputs, grad_inputs, in AddGradients()
1538 TF_AddGradients(graph_, outputs, noutputs, inputs, ninputs, nullptr, s_, in AddGradients()
/external/mesa3d/src/gallium/drivers/freedreno/ir3/
Dir3_group.c247 for (i = 0; i < ir->noutputs; i += 4) in find_neighbors()
250 for (i = 0; i < ir->noutputs; i++) { in find_neighbors()
Dir3_depth.c158 for (i = 0; i < ir->noutputs; i++) in ir3_depth()
Dir3_print.c236 for (unsigned i = 0; i < ir->noutputs; i++) { in ir3_print()
Dir3_compiler_nir.c2178 compile_assert(ctx, idx < ctx->ir->noutputs); in setup_output()
2196 unsigned ninputs, noutputs; in emit_instructions() local
2200 noutputs = (max_drvloc(&ctx->s->outputs) + 1) * 4; in emit_instructions()
2208 ctx->ir = ir3_create(ctx->compiler, ninputs, noutputs); in emit_instructions()
2422 ir->noutputs = j * 4; in ir3_compile_shader_nir()
2429 for (i = 0; i < ir->noutputs; i++) { in ir3_compile_shader_nir()
Dir3_cp.c572 for (unsigned i = 0; i < ir->noutputs; i++) { in ir3_cp()
Dir3_shader.c353 for (i = 0; i < ir->noutputs; i++) { in ir3_shader_disasm()
Dir3.h343 unsigned ninputs, noutputs; member
Dir3_ra.c884 for (unsigned i = 0; i < ir->noutputs; i++) { in ra_add_interference()
Dir3.c55 shader->noutputs = nout; in ir3_create()
/external/tensorflow/tensorflow/core/framework/
Dop_kernel.h572 OpKernelContext(Params* params, int noutputs);