Home
last modified time | relevance | path

Searched refs:ninputs (Results 1 – 18 of 18) 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.java279 int ninputs = inputHandles.length; in buildSubgraph() local
281 Output<?>[] inputs = new Output<?>[ninputs]; in buildSubgraph()
288 for (int i = 0; i < ninputs; i++) { in buildSubgraph()
323 int ninputs = inputs.length; in whileLoop() local
324 long[] inputHandles = new long[ninputs]; in whileLoop()
325 int[] inputIndices = new int[ninputs]; in whileLoop()
326 Output<?>[] outputs = new Output<?>[ninputs]; in whileLoop()
331 for (int i = 0; i < ninputs; i++) { in whileLoop()
339 for (int i = 0, j = ninputs; i < ninputs; ++i, ++j) { in whileLoop()
/external/tensorflow/tensorflow/c/
Dc_api.h1116 const int ninputs; member
1156 int ninputs,
1313 const TF_Operation* const* opers, int ninputs, const TF_Output* inputs,
1333 const TF_Operation* const* opers, int ninputs, const TF_Output* inputs,
1481 const TF_Output* inputs, TF_Tensor* const* input_values, int ninputs,
1503 const TF_Output* inputs, int ninputs,
1518 const TF_Output* inputs, TF_Tensor* const* input_values, int ninputs,
1558 int ninputs, const char** output_names,
1565 const char** input_names, int ninputs,
1574 int ninputs, const char** output_names,
Dwhile_loop_test.cc36 void Init(int ninputs) { in Init() argument
38 DCHECK_GT(ninputs, 0); in Init()
40 for (int i = 0; i < ninputs; ++i) { in Init()
58 outputs_.resize(ninputs, {nullptr, -1}); in Init()
145 EXPECT_EQ(params_->ninputs, 2); in TEST_F()
184 for (int i = 0; i < params_->ninputs; ++i) { in TEST_F()
Dc_api.cc825 const int ninputs = input_pairs->size(); in TF_Run_Inputs() local
826 for (int i = 0; i < ninputs; ++i) { in TF_Run_Inputs()
894 const char** c_input_names, TF_Tensor** c_inputs, int ninputs, in TF_Run() argument
901 std::vector<std::pair<string, Tensor>> input_pairs(ninputs); in TF_Run()
903 for (int i = 0; i < ninputs; ++i) { in TF_Run()
920 const char** c_input_names, int ninputs, in TF_PRunSetup() argument
928 std::vector<string> input_names(ninputs); in TF_PRunSetup()
931 for (int i = 0; i < ninputs; ++i) { in TF_PRunSetup()
952 const char** c_input_names, TF_Tensor** c_inputs, int ninputs, in TF_PRun() argument
959 std::vector<std::pair<string, Tensor>> input_pairs(ninputs); in TF_PRun()
[all …]
Dc_api_function.cc461 const TF_Graph* fn_body, const char* fn_name, int ninputs, in ProcessInputs() argument
465 input_tensors->reserve(ninputs); in ProcessInputs()
466 for (int i = 0; i < ninputs; ++i) { in ProcessInputs()
563 const TF_Operation* const* opers, int ninputs, const TF_Output* inputs, in TF_GraphToFunctionWithControlOutputs() argument
573 status->status = tensorflow::ProcessInputs(fn_body, fn_name, ninputs, inputs, in TF_GraphToFunctionWithControlOutputs()
630 int ninputs, const TF_Output* inputs, in TF_GraphToFunction() argument
636 fn_body, fn_name, append_hash_to_fn_name, num_opers, opers, ninputs, in TF_GraphToFunction()
Dc_api_test.cc1608 TF_Output* inputs, int ninputs, TF_Output* outputs, in AddGradients() argument
1617 ninputs, grad_inputs, s_, grad_outputs); in AddGradients()
1620 ninputs, nullptr, s_, grad_outputs); in AddGradients()
/external/pdfium/core/fpdfapi/page/
Dcpdf_function.cpp105 uint32_t ninputs, in Call() argument
108 if (m_nInputs != ninputs) in Call()
Dcpdf_function.h33 uint32_t ninputs,
/external/mesa3d/src/gallium/drivers/freedreno/ir3/
Dir3_compiler_nir.c1913 ctx->ir->ninputs = MAX2(ctx->ir->ninputs, r + 1); in add_sysval_input_compmask()
1985 create_collect(b, ctx->ir->inputs, ctx->ir->ninputs); in emit_intrinsic()
1989 dst[i] = create_indirect_load(ctx, ctx->ir->ninputs, in emit_intrinsic()
2972 compile_assert(ctx, idx < ctx->ir->ninputs); in setup_input()
2979 compile_assert(ctx, idx < ctx->ir->ninputs); in setup_input()
3083 unsigned ninputs, noutputs; in emit_instructions() local
3086 ninputs = (max_drvloc(&ctx->s->inputs) + 1) * 4; in emit_instructions()
3091 ninputs += max_sysvals[ctx->so->type]; in emit_instructions()
3093 ctx->ir = ir3_create(ctx->compiler, ninputs, noutputs); in emit_instructions()
3100 ninputs -= max_sysvals[ctx->so->type]; in emit_instructions()
[all …]
Dir3_depth.c190 for (i = 0; i < ir->ninputs; i++) { in ir3_depth()
Dir3_group.c245 for (i = 0; i < ir->ninputs; i += 4) in find_neighbors()
Dir3_ra.c1088 if (ctx->frag_face && (i < ir->ninputs) && ir->inputs[i]) { in ra_alloc()
1100 for (; i < ir->ninputs; i++) { in ra_alloc()
Dir3_shader.c393 for (i = 0; i < ir->ninputs; i++) { in ir3_shader_disasm()
Dir3.h406 unsigned ninputs, noutputs; member
Dir3.c52 shader->ninputs = nin; in ir3_create()
/external/tensorflow/tensorflow/python/ops/
Daccumulate_n_benchmark.py120 def _RunBenchmark(self, tag, input_fn, sizes, ninputs, repeats): argument
122 for ninput in ninputs: