Home
last modified time | relevance | path

Searched defs:elem_count (Results 1 – 23 of 23) sorted by relevance

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_fft.cc158 GpuExecutor *parent, Stream *stream, int rank, uint64 *elem_count, in Initialize()
319 int rank, uint64 *elem_count, in Initialize()
375 uint64 elem_count[1] = {num_x}; in Create1dPlan() local
391 uint64 elem_count[1] = {num_x}; in Create1dPlanWithScratchAllocator() local
406 uint64 elem_count[2] = {num_x, num_y}; in Create2dPlan() local
420 uint64 elem_count[2] = {num_x, num_y}; in Create2dPlanWithScratchAllocator() local
436 uint64 elem_count[3] = {num_x, num_y, num_z}; in Create3dPlan() local
450 uint64 elem_count[3] = {num_x, num_y, num_z}; in Create3dPlanWithScratchAllocator() local
462 Stream *stream, int rank, uint64 *elem_count, uint64 *input_embed, in CreateBatchedPlan()
480 Stream *stream, int rank, uint64 *elem_count, uint64 *input_embed, in CreateBatchedPlanWithScratchAllocator()
Drocm_blas.cc444 bool ROCMBlas::DoBlasAsum(Stream *stream, uint64 elem_count, in DoBlasAsum()
452 bool ROCMBlas::DoBlasAsum(Stream *stream, uint64 elem_count, in DoBlasAsum()
460 bool ROCMBlas::DoBlasAsum(Stream *stream, uint64 elem_count, in DoBlasAsum()
468 bool ROCMBlas::DoBlasAsum(Stream *stream, uint64 elem_count, in DoBlasAsum()
476 bool ROCMBlas::DoBlasAxpy(Stream *stream, uint64 elem_count, float alpha, in DoBlasAxpy()
485 bool ROCMBlas::DoBlasAxpy(Stream *stream, uint64 elem_count, double alpha, in DoBlasAxpy()
494 bool ROCMBlas::DoBlasAxpy(Stream *stream, uint64 elem_count, in DoBlasAxpy()
503 bool ROCMBlas::DoBlasAxpy(Stream *stream, uint64 elem_count, in DoBlasAxpy()
512 bool ROCMBlas::DoBlasCopy(Stream *stream, uint64 elem_count, in DoBlasCopy()
520 bool ROCMBlas::DoBlasCopy(Stream *stream, uint64 elem_count, in DoBlasCopy()
[all …]
/external/tensorflow/tensorflow/stream_executor/cuda/
Dcuda_fft.cc78 GpuExecutor *parent, Stream *stream, int rank, uint64 *elem_count, in Initialize()
236 int rank, uint64 *elem_count, in Initialize()
298 uint64 elem_count[1] = {num_x}; in Create1dPlan() local
315 uint64 elem_count[1] = {num_x}; in Create1dPlanWithScratchAllocator() local
331 uint64 elem_count[2] = {num_x, num_y}; in Create2dPlan() local
346 uint64 elem_count[2] = {num_x, num_y}; in Create2dPlanWithScratchAllocator() local
363 uint64 elem_count[3] = {num_x, num_y, num_z}; in Create3dPlan() local
379 uint64 elem_count[3] = {num_x, num_y, num_z}; in Create3dPlanWithScratchAllocator() local
393 Stream *stream, int rank, uint64 *elem_count, uint64 *input_embed, in CreateBatchedPlan()
420 Stream *stream, int rank, uint64 *elem_count, uint64 *input_embed, in CreateBatchedPlanWithScratchAllocator()
Dcuda_blas.cc577 bool CUDABlas::DoBlasAsum(Stream *stream, uint64 elem_count, in DoBlasInternalImpl()
585 bool CUDABlas::DoBlasAsum(Stream *stream, uint64 elem_count, in DoBlasInternalImpl()
593 bool CUDABlas::DoBlasAsum(Stream *stream, uint64 elem_count, in DoBlasInternalImpl()
601 bool CUDABlas::DoBlasAsum(Stream *stream, uint64 elem_count, in DoBlasInternalImpl()
609 bool CUDABlas::DoBlasAxpy(Stream *stream, uint64 elem_count, float alpha, in DoBlasInternalImpl()
617 bool CUDABlas::DoBlasAxpy(Stream *stream, uint64 elem_count, double alpha, in DoBlasInternalImpl()
625 bool CUDABlas::DoBlasAxpy(Stream *stream, uint64 elem_count, in DoBlasInternalImpl()
636 bool CUDABlas::DoBlasAxpy(Stream *stream, uint64 elem_count, in DoBlasInternalImpl()
647 bool CUDABlas::DoBlasCopy(Stream *stream, uint64 elem_count, in DoBlasInternalImpl()
655 bool CUDABlas::DoBlasCopy(Stream *stream, uint64 elem_count, in DoBlasInternalImpl()
[all …]
/external/skqp/src/compute/skc/platforms/cl_12/
Dconfig_cl.h95 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/llvm-project/libcxxabi/test/
Dcxa_vec_new_overflow_PR41395.pass.cpp39 const size_t elem_count = std::size_t(1) << (sizeof(std::size_t) * 8 - 2); in test_overflow_in_multiplication() local
77 const size_t elem_count = static_cast<size_t>(-1) / 4u; in test_overflow_in_addition() local
/external/llvm-project/lldb/include/lldb/Interpreter/
DOptionGroupValueObjectDisplay.h52 uint32_t elem_count; variable
/external/tensorflow/tensorflow/stream_executor/
Dstream.cc1467 Stream &Stream::ThenBlasAsum(uint64 elem_count, const DeviceMemory<float> &x, in ThenBlasAsum()
1477 Stream &Stream::ThenBlasAsum(uint64 elem_count, const DeviceMemory<double> &x, in ThenBlasAsum()
1488 Stream &Stream::ThenBlasAsum(uint64 elem_count, in ThenBlasAsum()
1500 Stream &Stream::ThenBlasAsum(uint64 elem_count, in ThenBlasAsum()
1512 Stream &Stream::ThenBlasAxpy(uint64 elem_count, float alpha, in ThenBlasAxpy()
1525 Stream &Stream::ThenBlasAxpy(uint64 elem_count, double alpha, in ThenBlasAxpy()
1538 Stream &Stream::ThenBlasAxpy(uint64 elem_count, std::complex<float> alpha, in ThenBlasAxpy()
1553 Stream &Stream::ThenBlasAxpy(uint64 elem_count, std::complex<double> alpha, in ThenBlasAxpy()
1568 Stream &Stream::ThenBlasCopy(uint64 elem_count, const DeviceMemory<float> &x, in ThenBlasCopy()
1579 Stream &Stream::ThenBlasCopy(uint64 elem_count, const DeviceMemory<double> &x, in ThenBlasCopy()
[all …]
Ddevice_memory.h178 explicit SharedDeviceMemory(uint64 elem_count) in SharedDeviceMemory()
/external/python/cpython3/Python/
Dast_unparse.c332 Py_ssize_t i, elem_count; in append_ast_set() local
347 Py_ssize_t i, elem_count; in append_ast_list() local
362 Py_ssize_t i, elem_count; in append_ast_tuple() local
722 Py_ssize_t i, elem_count; in append_ast_constant() local
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_import.cc322 int elem_count = bytes_len / read_size; in ReadAsLittleEndian() local
359 int elem_count = bytes_len / 2; in ConvertFloatBuffer() local
378 int elem_count = bytes_len / 4; in ConvertFloatBuffer() local
394 int elem_count = bytes_len / 8; in ConvertFloatBuffer() local
/external/rust/crates/grpcio-sys/grpc/src/core/lib/iomgr/
Dcombiner.cc267 #define OLD_STATE_WAS(orphaned, elem_count) \ in grpc_combiner_continue_exec_ctx() argument
/external/grpc-grpc/src/core/lib/iomgr/
Dcombiner.cc290 #define OLD_STATE_WAS(orphaned, elem_count) \ in grpc_combiner_continue_exec_ctx() argument
/external/deqp-deps/amber/src/
Dverifier.cc529 size_t elem_count = values.size() / fmt->InputNeededPerElement(); in ProbeSSBO() local
/external/gemmlowp/meta/generators/
Dneon_emitter.py413 elem_count, argument
537 elem_count, argument
666 def EmitVSumReduce(self, reduce_type, elem_count, reduce_count, destinations, argument
Dneon_emitter_64.py839 elem_count, argument
955 elem_count, argument
1057 def EmitVSumReduce(self, reduce_type, elem_count, reduce_count, destinations, argument
/external/wpa_supplicant_8/src/rsn_supp/
Dwpa_ft.c193 u8 *buf, *pos, *ftie_len, *ftie_pos, *fte_mic, *elem_count; in wpa_ft_gen_req_ies() local
/external/mesa3d/src/intel/vulkan/
DgenX_pipeline.c105 uint32_t elem_count = __builtin_popcount(elements) - in emit_vertex_input() local
/external/tensorflow/tensorflow/compiler/xla/service/
Dhlo_graph_dumper.cc842 optional<int64> elem_count; in GetInstructionNodeInlinedOperands() local
/external/mesa3d/src/amd/vulkan/
Dradv_query.c369 nir_ssa_def *elem_count = nir_ushr(&b, stats_mask, nir_imm_int(&b, 16)); in build_pipeline_statistics_query_shader() local
/external/mesa3d/src/compiler/spirv/
Dspirv_to_nir.c1905 unsigned elem_count = count - 3; in vtn_handle_constant() local
/external/mesa3d/src/amd/llvm/
Dac_nir_to_llvm.c2359 LLVMValueRef elem_count = in get_image_buffer_descriptor() local
/external/wpa_supplicant_8/src/ap/
Dwpa_auth_ft.c2511 u8 *fte_mic, *elem_count; in wpa_sm_write_assoc_resp_ies() local