| /external/tensorflow/tensorflow/stream_executor/cuda/ |
| D | cuda_fft.cc | 77 GpuExecutor *parent, Stream *stream, int rank, uint64 *elem_count, in Initialize()
234 int rank, uint64 *elem_count, in Initialize()
295 uint64 elem_count[1] = {num_x}; in Create1dPlan() local
312 uint64 elem_count[1] = {num_x}; in Create1dPlanWithScratchAllocator() local
328 uint64 elem_count[2] = {num_x, num_y}; in Create2dPlan() local
343 uint64 elem_count[2] = {num_x, num_y}; in Create2dPlanWithScratchAllocator() local
360 uint64 elem_count[3] = {num_x, num_y, num_z}; in Create3dPlan() local
376 uint64 elem_count[3] = {num_x, num_y, num_z}; in Create3dPlanWithScratchAllocator() local
390 Stream *stream, int rank, uint64 *elem_count, uint64 *input_embed, in CreateBatchedPlan()
417 Stream *stream, int rank, uint64 *elem_count, uint64 *input_embed, in CreateBatchedPlanWithScratchAllocator()
|
| D | cuda_blas.cc | 433 bool CUDABlas::DoBlasAsum(Stream *stream, uint64 elem_count, in DoBlasAsum()
441 bool CUDABlas::DoBlasAsum(Stream *stream, uint64 elem_count, in DoBlasAsum()
449 bool CUDABlas::DoBlasAsum(Stream *stream, uint64 elem_count, in DoBlasAsum()
457 bool CUDABlas::DoBlasAsum(Stream *stream, uint64 elem_count, in DoBlasAsum()
465 bool CUDABlas::DoBlasAxpy(Stream *stream, uint64 elem_count, float alpha, in DoBlasAxpy()
473 bool CUDABlas::DoBlasAxpy(Stream *stream, uint64 elem_count, double alpha, in DoBlasAxpy()
481 bool CUDABlas::DoBlasAxpy(Stream *stream, uint64 elem_count, in DoBlasAxpy()
491 bool CUDABlas::DoBlasAxpy(Stream *stream, uint64 elem_count, in DoBlasAxpy()
501 bool CUDABlas::DoBlasCopy(Stream *stream, uint64 elem_count, in DoBlasCopy()
509 bool CUDABlas::DoBlasCopy(Stream *stream, uint64 elem_count, in DoBlasCopy()
[all …]
|
| /external/tensorflow/tensorflow/stream_executor/rocm/ |
| D | rocm_fft.cc | 146 GpuExecutor *parent, Stream *stream, int rank, uint64 *elem_count, in Initialize()
339 int rank, uint64 *elem_count, in Initialize()
376 uint64 elem_count[1] = {num_x}; in Create1dPlan() local
392 uint64 elem_count[1] = {num_x}; in Create1dPlanWithScratchAllocator() local
407 uint64 elem_count[2] = {num_x, num_y}; in Create2dPlan() local
421 uint64 elem_count[2] = {num_x, num_y}; in Create2dPlanWithScratchAllocator() local
437 uint64 elem_count[3] = {num_x, num_y, num_z}; in Create3dPlan() local
451 uint64 elem_count[3] = {num_x, num_y, num_z}; in Create3dPlanWithScratchAllocator() local
463 Stream *stream, int rank, uint64 *elem_count, uint64 *input_embed, in CreateBatchedPlan()
481 Stream *stream, int rank, uint64 *elem_count, uint64 *input_embed, in CreateBatchedPlanWithScratchAllocator()
|
| D | rocm_blas.cc | 400 bool ROCMBlas::DoBlasAsum(Stream *stream, uint64 elem_count, in DoBlasAsum()
408 bool ROCMBlas::DoBlasAsum(Stream *stream, uint64 elem_count, in DoBlasAsum()
416 bool ROCMBlas::DoBlasAsum(Stream *stream, uint64 elem_count, in DoBlasAsum()
424 bool ROCMBlas::DoBlasAsum(Stream *stream, uint64 elem_count, in DoBlasAsum()
432 bool ROCMBlas::DoBlasAxpy(Stream *stream, uint64 elem_count, float alpha, in DoBlasAxpy()
440 bool ROCMBlas::DoBlasAxpy(Stream *stream, uint64 elem_count, double alpha, in DoBlasAxpy()
448 bool ROCMBlas::DoBlasAxpy(Stream *stream, uint64 elem_count, in DoBlasAxpy()
457 bool ROCMBlas::DoBlasAxpy(Stream *stream, uint64 elem_count, in DoBlasAxpy()
466 bool ROCMBlas::DoBlasCopy(Stream *stream, uint64 elem_count, in DoBlasCopy()
474 bool ROCMBlas::DoBlasCopy(Stream *stream, uint64 elem_count, in DoBlasCopy()
[all …]
|
| /external/skqp/src/compute/skc/platforms/cl_12/ |
| D | config_cl.h | 95 skc_uint elem_count; member
100 skc_uint elem_count; member
105 skc_uint elem_count; member
110 skc_uint elem_count; member
115 skc_uint elem_count; member
130 skc_uint elem_count; member
135 skc_uint elem_count; member
139 skc_uint elem_count; member
|
| /external/skia/src/compute/skc/platforms/cl_12/ |
| D | config_cl.h | 95 skc_uint elem_count; member
100 skc_uint elem_count; member
105 skc_uint elem_count; member
110 skc_uint elem_count; member
115 skc_uint elem_count; member
130 skc_uint elem_count; member
135 skc_uint elem_count; member
139 skc_uint elem_count; member
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | stream.cc | 1915 Stream &Stream::ThenBlasAsum(uint64 elem_count, const DeviceMemory<float> &x, in ThenBlasAsum()
1925 Stream &Stream::ThenBlasAsum(uint64 elem_count, const DeviceMemory<double> &x, in ThenBlasAsum()
1935 Stream &Stream::ThenBlasAsum(uint64 elem_count, in ThenBlasAsum()
1946 Stream &Stream::ThenBlasAsum(uint64 elem_count, in ThenBlasAsum()
1957 Stream &Stream::ThenBlasAxpy(uint64 elem_count, float alpha, in ThenBlasAxpy()
1969 Stream &Stream::ThenBlasAxpy(uint64 elem_count, double alpha, in ThenBlasAxpy()
1981 Stream &Stream::ThenBlasAxpy(uint64 elem_count, std::complex<float> alpha, in ThenBlasAxpy()
1995 Stream &Stream::ThenBlasAxpy(uint64 elem_count, std::complex<double> alpha, in ThenBlasAxpy()
2009 Stream &Stream::ThenBlasCopy(uint64 elem_count, const DeviceMemory<float> &x, in ThenBlasCopy()
2019 Stream &Stream::ThenBlasCopy(uint64 elem_count, const DeviceMemory<double> &x, in ThenBlasCopy()
[all …]
|
| D | device_memory.h | 173 explicit SharedDeviceMemory(uint64 elem_count) in SharedDeviceMemory()
|
| /external/webrtc/webrtc/libjingle/xmllite/ |
| D | xmlelement_unittest.cc | 244 int elem_count = 100; // Was 100000, but that's too slow. in TEST() local
|
| /external/python/cpython3/Python/ |
| D | ast_unparse.c | 304 Py_ssize_t i, elem_count; in append_ast_set() local
319 Py_ssize_t i, elem_count; in append_ast_list() local
334 Py_ssize_t i, elem_count; in append_ast_tuple() local
|
| /external/grpc-grpc/src/core/lib/iomgr/ |
| D | combiner.cc | 290 #define OLD_STATE_WAS(orphaned, elem_count) \ in grpc_combiner_continue_exec_ctx() argument
|
| /external/wpa_supplicant_8/src/rsn_supp/ |
| D | wpa_ft.c | 160 u8 *buf, *pos, *ftie_len, *ftie_pos, *fte_mic, *elem_count; in wpa_ft_gen_req_ies() local
|
| /external/gemmlowp/meta/generators/ |
| D | neon_emitter.py | 413 elem_count, argument
537 elem_count, argument
666 def EmitVSumReduce(self, reduce_type, elem_count, reduce_count, destinations, argument
|
| D | neon_emitter_64.py | 839 elem_count, argument
955 elem_count, argument
1057 def EmitVSumReduce(self, reduce_type, elem_count, reduce_count, destinations, argument
|
| /external/mesa3d/src/intel/vulkan/ |
| D | genX_pipeline.c | 103 uint32_t elem_count = __builtin_popcount(elements) - in emit_vertex_input() local
|
| /external/mesa3d/src/amd/vulkan/ |
| D | radv_query.c | 376 nir_ssa_def *elem_count = nir_ushr(&b, stats_mask, nir_imm_int(&b, 16)); in build_pipeline_statistics_query_shader() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | hlo_graph_dumper.cc | 828 optional<int64> elem_count; in GetInstructionNodeInlinedOperands() local
|
| /external/mesa3d/src/amd/common/ |
| D | ac_llvm_build.c | 1037 …LLVMValueRef elem_count = LLVMBuildExtractElement(ctx->builder, rsrc, LLVMConstInt(ctx->i32, 2, 0)… in ac_build_buffer_load_format_gfx9_safe() local
|
| D | ac_nir_to_llvm.c | 3625 …LLVMValueRef elem_count = LLVMBuildExtractElement(ctx->ac.builder, rsrc, LLVMConstInt(ctx->ac.i32,… in get_image_buffer_descriptor() local
|
| /external/mesa3d/src/compiler/spirv/ |
| D | spirv_to_nir.c | 1452 unsigned elem_count = count - 3; in vtn_handle_constant() local
|
| /external/wpa_supplicant_8/src/ap/ |
| D | wpa_auth_ft.c | 2412 u8 *fte_mic, *elem_count; in wpa_sm_write_assoc_resp_ies() local
|