Home
last modified time | relevance | path

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

12

/external/tensorflow/tensorflow/stream_executor/rocm/
Drocm_fft.cc158 GpuExecutor *parent, Stream *stream, int rank, uint64_t *elem_count, in Initialize()
320 int rank, uint64_t *elem_count, in Initialize()
377 uint64_t elem_count[1] = {num_x}; in Create1dPlan() local
393 uint64_t elem_count[1] = {num_x}; in Create1dPlanWithScratchAllocator() local
408 uint64_t elem_count[2] = {num_x, num_y}; in Create2dPlan() local
422 uint64_t elem_count[2] = {num_x, num_y}; in Create2dPlanWithScratchAllocator() local
438 uint64_t elem_count[3] = {num_x, num_y, num_z}; in Create3dPlan() local
452 uint64_t elem_count[3] = {num_x, num_y, num_z}; in Create3dPlanWithScratchAllocator() local
464 Stream *stream, int rank, uint64_t *elem_count, uint64 *input_embed, in CreateBatchedPlan()
482 Stream *stream, int rank, uint64_t *elem_count, uint64 *input_embed, in CreateBatchedPlanWithScratchAllocator()
Drocm_blas.cc206 bool ROCMBlas::DoBlasAxpy(Stream *stream, uint64_t elem_count, float alpha, in DoBlasAxpy()
215 bool ROCMBlas::DoBlasAxpy(Stream *stream, uint64_t elem_count, double alpha, in DoBlasAxpy()
224 bool ROCMBlas::DoBlasAxpy(Stream *stream, uint64_t elem_count, in DoBlasAxpy()
233 bool ROCMBlas::DoBlasAxpy(Stream *stream, uint64_t elem_count, in DoBlasAxpy()
242 bool ROCMBlas::DoBlasCopy(Stream *stream, uint64_t elem_count, in DoBlasCopy()
250 bool ROCMBlas::DoBlasCopy(Stream *stream, uint64_t elem_count, in DoBlasCopy()
258 bool ROCMBlas::DoBlasCopy(Stream *stream, uint64_t elem_count, in DoBlasCopy()
266 bool ROCMBlas::DoBlasCopy(Stream *stream, uint64_t elem_count, in DoBlasCopy()
274 bool ROCMBlas::DoBlasScal(Stream *stream, uint64_t elem_count, float alpha, in DoBlasScal()
282 bool ROCMBlas::DoBlasScal(Stream *stream, uint64_t elem_count, double alpha, in DoBlasScal()
[all …]
/external/tensorflow/tensorflow/compiler/xla/stream_executor/cuda/
Dcuda_fft.cc97 GpuExecutor *parent, Stream *stream, int rank, uint64_t *elem_count, in Initialize()
257 int rank, uint64_t *elem_count, in Initialize()
318 uint64_t elem_count[1] = {num_x}; in Create1dPlan() local
334 uint64_t elem_count[1] = {num_x}; in Create1dPlanWithScratchAllocator() local
351 uint64_t elem_count[2] = {num_x, num_y}; in Create2dPlan() local
367 uint64_t elem_count[2] = {num_x, num_y}; in Create2dPlanWithScratchAllocator() local
385 uint64_t elem_count[3] = {num_x, num_y, num_z}; in Create3dPlan() local
402 uint64_t elem_count[3] = {num_x, num_y, num_z}; in Create3dPlanWithScratchAllocator() local
417 Stream *stream, int rank, uint64_t *elem_count, uint64 *input_embed, in CreateBatchedPlan()
445 Stream *stream, int rank, uint64_t *elem_count, uint64 *input_embed, in CreateBatchedPlanWithScratchAllocator()
Dcuda_blas.cc422 bool CUDABlas::DoBlasAxpy(Stream *stream, uint64_t elem_count, float alpha, in DoBlasInternalImpl()
430 bool CUDABlas::DoBlasAxpy(Stream *stream, uint64_t elem_count, double alpha, in DoBlasInternalImpl()
438 bool CUDABlas::DoBlasAxpy(Stream *stream, uint64_t elem_count, in DoBlasInternalImpl()
449 bool CUDABlas::DoBlasAxpy(Stream *stream, uint64_t elem_count, in DoBlasInternalImpl()
460 bool CUDABlas::DoBlasCopy(Stream *stream, uint64_t elem_count, in DoBlasInternalImpl()
468 bool CUDABlas::DoBlasCopy(Stream *stream, uint64_t elem_count, in DoBlasInternalImpl()
476 bool CUDABlas::DoBlasCopy(Stream *stream, uint64_t elem_count, in DoBlasInternalImpl()
484 bool CUDABlas::DoBlasCopy(Stream *stream, uint64_t elem_count, in DoBlasInternalImpl()
492 bool CUDABlas::DoBlasScal(Stream *stream, uint64_t elem_count, float alpha, in DoBlasInternalImpl()
498 bool CUDABlas::DoBlasScal(Stream *stream, uint64_t elem_count, double alpha, in DoBlasInternalImpl()
[all …]
/external/cronet/buildtools/third_party/libc++abi/trunk/test/
Dcxa_vec_new_overflow_PR41395.pass.cpp42 const size_t elem_count = std::size_t(1) << (sizeof(std::size_t) * 8 - 2); in test_overflow_in_multiplication() local
80 const size_t elem_count = static_cast<size_t>(-1) / 4u; in test_overflow_in_addition() local
/external/python/cpython3/Python/
Dast_unparse.c337 Py_ssize_t i, elem_count; in append_ast_set() local
352 Py_ssize_t i, elem_count; in append_ast_list() local
367 Py_ssize_t i, elem_count; in append_ast_tuple() local
727 Py_ssize_t i, elem_count; in append_ast_constant() local
/external/tensorflow/tensorflow/compiler/xla/stream_executor/
Dstream.cc1099 Stream &Stream::ThenBlasAxpy(uint64_t elem_count, float alpha, in ThenBlasAxpy()
1112 Stream &Stream::ThenBlasAxpy(uint64_t elem_count, double alpha, in ThenBlasAxpy()
1125 Stream &Stream::ThenBlasAxpy(uint64_t elem_count, std::complex<float> alpha, in ThenBlasAxpy()
1140 Stream &Stream::ThenBlasAxpy(uint64_t elem_count, std::complex<double> alpha, in ThenBlasAxpy()
1155 Stream &Stream::ThenBlasCopy(uint64_t elem_count, const DeviceMemory<float> &x, in ThenBlasCopy()
1166 Stream &Stream::ThenBlasCopy(uint64_t elem_count, const DeviceMemory<double> &x, in ThenBlasCopy()
1177 Stream &Stream::ThenBlasCopy(uint64_t elem_count, in ThenBlasCopy()
1190 Stream &Stream::ThenBlasCopy(uint64_t elem_count, in ThenBlasCopy()
1203 Stream &Stream::ThenBlasScal(uint64_t elem_count, float alpha, in ThenBlasScal()
1211 Stream &Stream::ThenBlasScal(uint64_t elem_count, double alpha, in ThenBlasScal()
[all …]
Ddevice_memory.h178 explicit SharedDeviceMemory(uint64_t elem_count) in SharedDeviceMemory()
/external/rust/crates/grpcio-sys/grpc/src/core/lib/iomgr/
Dcombiner.cc268 #define OLD_STATE_WAS(orphaned, elem_count) \ in grpc_combiner_continue_exec_ctx() argument
/external/tensorflow/tensorflow/compiler/mlir/lite/
Dflatbuffer_import.cc342 int elem_count = bytes_len / read_size; in ReadAsLittleEndian() local
387 int elem_count = bytes_len / 2; in ConvertFloatBuffer() local
406 int elem_count = bytes_len / 4; in ConvertFloatBuffer() local
423 int elem_count = bytes_len / 8; in ConvertFloatBuffer() local
/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/pdfium/testing/fuzzers/
Dpdf_xfa_fdp_fuzzer.cc860 int elem_count = data_provider->ConsumeIntegralInRange(1, 6); in GenXfaTree() 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.c211 u8 *buf, *pos, *ftie_len, *ftie_pos, *fte_mic, *elem_count; in wpa_ft_gen_req_ies() local
/external/wpa_supplicant_8/src/ap/
Dbeacon.c472 u8 elem_count = 0, *elem = NULL, **elem_offset = NULL, *end; in ieee802_11_build_ap_params_mbssid() local
Dieee802_11.c6886 u8 *elem_count, const u8 *known_bss, in hostapd_eid_mbssid_len()
6935 size_t *bss_index, u8 elem_count, in hostapd_eid_mbssid_elem()
7038 unsigned int frame_stype, u8 elem_count, in hostapd_eid_mbssid()
/external/mesa3d/src/intel/vulkan/
DgenX_pipeline.c105 uint32_t elem_count = __builtin_popcount(elements) - in emit_vertex_input() 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/tensorflow/tensorflow/compiler/xla/service/
Dhlo_graph_dumper.cc880 optional<int64_t> elem_count; in GetInstructionNodeInlinedOperands() local
/external/tensorflow/tensorflow/compiler/xla/tests/
Dvalue_inference_test.cc706 int64_t elem_count = 17; in TEST_F() local
/external/mesa3d/src/compiler/spirv/
Dspirv_to_nir.c1905 unsigned elem_count = count - 3; in vtn_handle_constant() local
/external/sg3_utils/src/
Dsg_ses.c3600 int j, k, desc_len, etype, el_num, ind, elem_count, ei, eiioe, num_elems; in additional_elem_sdg() local
/external/mesa3d/src/amd/llvm/
Dac_nir_to_llvm.c2359 LLVMValueRef elem_count = in get_image_buffer_descriptor() local

12