| /external/tensorflow/tensorflow/core/kernels/ |
| D | in_topk_op_gpu.cu.cc | 43 int num_targets, int num_classes) { in ComputePredictionMaskKernel() argument
44 GPU_1D_KERNEL_LOOP(i, num_targets * num_classes) { in ComputePredictionMaskKernel()
95 const Eigen::Index num_targets = predictions.dimension(0); in operator ()() local
99 context, num_targets * num_classes < std::numeric_limits<int>::max(), in operator ()()
103 if (num_targets == 0 || num_classes == 0) { in operator ()()
113 TensorShape({num_targets, num_classes}), in operator ()()
121 context->allocate_temp(DT_INT64, TensorShape({num_targets}), in operator ()()
127 GpuLaunchConfig config = GetGpuLaunchConfig(num_targets * num_classes, d); in operator ()()
133 num_targets, num_classes)); in operator ()()
156 Dims<1>(num_targets))); in operator ()()
[all …]
|
| D | in_topk_op_test.cc | 36 static Graph* InTopK(int num_targets, int num_classes, T top_k) { in InTopK() argument
41 Tensor predictions_t(DT_FLOAT, TensorShape({num_targets, num_classes})); in InTopK()
44 Tensor targets_t(dtype, TensorShape({num_targets})); in InTopK()
|
| D | in_topk_op.h | 59 const Eigen::Index num_targets = predictions.dimension(0);
71 for (int batch_idx = 0; batch_idx < num_targets; batch_idx++) {
|
| /external/mesa3d/src/gallium/drivers/radeonsi/ |
| D | si_state_streamout.c | 86 static void si_set_streamout_targets(struct pipe_context *ctx, unsigned num_targets, in si_set_streamout_targets() argument
91 unsigned old_num_targets = sctx->streamout.num_targets; in si_set_streamout_targets()
96 if (sctx->streamout.num_targets && sctx->streamout.begin_emitted) { in si_set_streamout_targets()
106 for (i = 0; i < sctx->streamout.num_targets; i++) in si_set_streamout_targets()
141 if (num_targets) { in si_set_streamout_targets()
154 if (sctx->streamout.num_targets && sctx->streamout.begin_emitted) in si_set_streamout_targets()
159 for (i = 0; i < num_targets; i++) { in si_set_streamout_targets()
171 for (; i < sctx->streamout.num_targets; i++) in si_set_streamout_targets()
175 sctx->streamout.num_targets = num_targets; in si_set_streamout_targets()
179 if (num_targets) { in si_set_streamout_targets()
[all …]
|
| /external/mesa3d/src/gallium/drivers/virgl/ |
| D | virgl_streamout.c | 73 unsigned num_targets, in virgl_set_so_targets() argument
79 for (i = 0; i < num_targets; i++) { in virgl_set_so_targets()
91 for (i = num_targets; i < vctx->num_so_targets; i++) in virgl_set_so_targets()
93 vctx->num_so_targets = num_targets; in virgl_set_so_targets()
94 virgl_encoder_set_so_targets(vctx, num_targets, targets, 0);//append_bitmask); in virgl_set_so_targets()
|
| /external/mesa3d/src/gallium/drivers/r600/ |
| D | r600_streamout.c | 114 unsigned num_targets, in r600_set_streamout_targets() argument
123 if (rctx->streamout.num_targets && rctx->streamout.begin_emitted) { in r600_set_streamout_targets()
128 for (i = 0; i < num_targets; i++) { in r600_set_streamout_targets()
138 for (; i < rctx->streamout.num_targets; i++) { in r600_set_streamout_targets()
144 rctx->streamout.num_targets = num_targets; in r600_set_streamout_targets()
147 if (num_targets) { in r600_set_streamout_targets()
190 for (i = 0; i < rctx->streamout.num_targets; i++) { in r600_emit_streamout_begin()
263 for (i = 0; i < rctx->streamout.num_targets; i++) { in r600_emit_streamout_end()
|
| /external/mesa3d/src/mesa/state_tracker/ |
| D | st_cb_xformfb.c | 54 unsigned num_targets; member
96 for (i = 0; i < sobj->num_targets; i++) { in st_delete_transform_feedback()
143 sobj->num_targets = i+1; in st_begin_transform_feedback()
150 cso_set_stream_outputs(st->cso_context, sobj->num_targets, in st_begin_transform_feedback()
177 cso_set_stream_outputs(st->cso_context, sobj->num_targets, in st_resume_transform_feedback()
|
| /external/mesa3d/src/gallium/drivers/softpipe/ |
| D | sp_state_so.c | 64 unsigned num_targets, in softpipe_set_so_targets() argument
71 for (i = 0; i < num_targets; i++) { in softpipe_set_so_targets()
84 softpipe->num_so_targets = num_targets; in softpipe_set_so_targets()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/cl/ |
| D | run_tests.sh | 81 num_targets=${#targets[@]}
82 if ((num_targets == 1)); then
93 for ((i = 0; i < num_targets; i++)); do
|
| /external/mesa3d/src/gallium/drivers/llvmpipe/ |
| D | lp_state_so.c | 65 unsigned num_targets, in llvmpipe_set_so_targets() argument
71 for (i = 0; i < num_targets; i++) { in llvmpipe_set_so_targets()
98 llvmpipe->num_so_targets = num_targets; in llvmpipe_set_so_targets()
|
| /external/mesa3d/src/gallium/auxiliary/draw/ |
| D | draw_pt_so_emit.c | 98 for (i = 0; i < draw->so.num_targets; ++i) { in draw_pt_so_emit_prepare()
135 for (i = 0; i < draw->so.num_targets; i++) { in so_emit_prim()
165 for (ob = 0; ob < draw->so.num_targets; ++ob) { in so_emit_prim()
219 for (ob = 0; ob < draw->so.num_targets; ++ob) { in so_emit_prim()
295 if (!draw->so.num_targets) in draw_pt_so_emit()
|
| D | draw_context.c | 1085 int num_targets, in draw_set_mapped_so_targets() argument
1092 for (i = 0; i < num_targets; i++) in draw_set_mapped_so_targets()
1094 for (i = num_targets; i < PIPE_MAX_SO_BUFFERS; i++) in draw_set_mapped_so_targets()
1097 draw->so.num_targets = num_targets; in draw_set_mapped_so_targets()
|
| /external/mesa3d/src/gallium/drivers/svga/ |
| D | svga_pipe_streamout.c | 417 unsigned num_targets, in svga_set_stream_output_targets() argument
425 boolean begin_so_queries = num_targets > 0; in svga_set_stream_output_targets()
428 num_targets); in svga_set_stream_output_targets()
448 assert(num_targets <= SVGA3D_DX_MAX_SOTARGETS); in svga_set_stream_output_targets()
450 for (i = 0; i < num_targets; i++) { in svga_set_stream_output_targets()
486 num_so_targets = MAX2(svga->num_so_targets, num_targets); in svga_set_stream_output_targets()
489 svga->num_so_targets = num_targets; in svga_set_stream_output_targets()
|
| /external/mesa3d/src/gallium/auxiliary/util/ |
| D | u_blitter.h | 554 unsigned num_targets, in util_blitter_save_so_targets() argument
558 assert(num_targets <= ARRAY_SIZE(blitter->saved_so_targets)); in util_blitter_save_so_targets()
560 blitter->saved_num_so_targets = num_targets; in util_blitter_save_so_targets()
561 for (i = 0; i < num_targets; i++) in util_blitter_save_so_targets()
|
| /external/mesa3d/src/gallium/drivers/freedreno/ |
| D | freedreno_state.c | 508 unsigned num_targets, struct pipe_stream_output_target **targets, in fd_set_stream_output_targets() argument
515 debug_assert(num_targets <= ARRAY_SIZE(so->targets)); in fd_set_stream_output_targets()
517 for (i = 0; i < num_targets; i++) { in fd_set_stream_output_targets()
532 for (; i < so->num_targets; i++) { in fd_set_stream_output_targets()
536 so->num_targets = num_targets; in fd_set_stream_output_targets()
|
| D | freedreno_draw.c | 194 for (unsigned i = 0; i < ctx->streamout.num_targets; i++) in batch_draw_tracking()
238 if (ctx->streamout.num_targets > 0) in fd_draw_vbo()
305 if (ctx->streamout.num_targets > 0) in fd_draw_vbo()
326 for (unsigned i = 0; i < ctx->streamout.num_targets; i++) in fd_draw_vbo()
|
| /external/mesa3d/src/gallium/drivers/v3d/ |
| D | v3d_query.c | 88 if (v3d->streamout.num_targets > 0) in v3d_begin_query()
130 if (v3d->streamout.num_targets > 0) in v3d_end_query()
|
| D | v3dx_state.c | 1212 unsigned num_targets, in v3d_set_stream_output_targets() argument
1220 assert(num_targets <= ARRAY_SIZE(so->targets)); in v3d_set_stream_output_targets()
1227 if (num_targets == 0 && so->num_targets > 0) in v3d_set_stream_output_targets()
1230 for (i = 0; i < num_targets; i++) { in v3d_set_stream_output_targets()
1237 for (; i < so->num_targets; i++) in v3d_set_stream_output_targets()
1240 so->num_targets = num_targets; in v3d_set_stream_output_targets()
1243 if (num_targets > 0 && !ctx->prim_counts) { in v3d_set_stream_output_targets()
|
| /external/mesa3d/src/gallium/drivers/panfrost/ |
| D | pan_context.c | 236 if (!ctx->streamout.num_targets) in panfrost_statistics_record()
245 for (unsigned i = 0; i < ctx->streamout.num_targets; ++i) { in panfrost_update_streamout_offsets()
1483 unsigned num_targets, in panfrost_set_stream_output_targets() argument
1490 assert(num_targets <= ARRAY_SIZE(so->targets)); in panfrost_set_stream_output_targets()
1492 for (unsigned i = 0; i < num_targets; i++) { in panfrost_set_stream_output_targets()
1499 for (unsigned i = 0; i < so->num_targets; i++) in panfrost_set_stream_output_targets()
1502 so->num_targets = num_targets; in panfrost_set_stream_output_targets()
|
| /external/deqp/external/openglcts/modules/gles31/ |
| D | es31cFramebufferNoAttachmentsTests.cpp | 232 unsigned num_targets = sizeof(targets) / sizeof(GLenum); in iterate() local
281 for (unsigned j = 0; j < num_targets; ++j) in iterate()
302 for (unsigned i = 0; i < num_targets; ++i) in iterate()
|
| /external/llvm-project/lldb/source/Plugins/DynamicLoader/Hexagon-DYLD/ |
| D | DynamicLoaderHexagonDYLD.cpp | 433 size_t num_targets = target_symbols.GetSize(); in GetStepThroughTrampolinePlan() local
434 if (!num_targets) in GetStepThroughTrampolinePlan()
439 for (size_t i = 0; i < num_targets; ++i) { in GetStepThroughTrampolinePlan()
|
| /external/tensorflow/tensorflow/core/distributed_runtime/ |
| D | message_wrappers.h | 77 virtual size_t num_targets() const = 0;
129 size_t num_targets() const override;
180 size_t num_targets() const override;
221 size_t num_targets() const override;
|
| D | message_wrappers.cc | 82 size_t InMemoryRunStepRequest::num_targets() const { return targets_.size(); } in num_targets() function in tensorflow::InMemoryRunStepRequest
124 for (size_t i = 0; i < num_targets(); ++i) { in ToProto()
186 size_t MutableProtoRunStepRequest::num_targets() const { in num_targets() function in tensorflow::MutableProtoRunStepRequest
265 size_t ProtoRunStepRequest::num_targets() const { in num_targets() function in tensorflow::ProtoRunStepRequest
|
| /external/mesa3d/src/gallium/auxiliary/driver_ddebug/ |
| D | dd_context.c | 567 unsigned num_targets, in dd_context_set_stream_output_targets() argument
575 dstate->num_so_targets = num_targets; in dd_context_set_stream_output_targets()
576 safe_memcpy(dstate->so_targets, tgs, sizeof(*tgs) * num_targets); in dd_context_set_stream_output_targets()
577 safe_memcpy(dstate->so_offsets, offsets, sizeof(*offsets) * num_targets); in dd_context_set_stream_output_targets()
578 pipe->set_stream_output_targets(pipe, num_targets, tgs, offsets); in dd_context_set_stream_output_targets()
|
| /external/llvm-project/lldb/source/Plugins/DynamicLoader/POSIX-DYLD/ |
| D | DynamicLoaderPOSIXDYLD.cpp | 475 size_t num_targets = target_symbols.GetSize(); in GetStepThroughTrampolinePlan() local
476 if (!num_targets) in GetStepThroughTrampolinePlan()
481 for (size_t i = 0; i < num_targets; ++i) { in GetStepThroughTrampolinePlan()
|