Home
last modified time | relevance | path

Searched defs:batch_id (Results 1 – 25 of 34) sorted by relevance

12

/aosp_15_r20/external/ComputeLibrary/src/cpu/kernels/
H A DCpuGemmLowpOffsetContributionOutputStageKernel.cpp537 const int batch_id = id.z() / depth_input; in run_offset_contribution_output_stage() local
555 const int batch_id = id.z() / depth_input; in run_offset_contribution_output_stage() local
581 const int batch_id = id.z() / depth_input; in run_offset_contribution_output_stage() local
597 const int batch_id = id.z() / depth_input; in run_offset_contribution_output_stage() local
623 const int batch_id = id.z() / depth_input; in run_offset_contribution_output_stage() local
638 const int batch_id = id.z() / depth_input; in run_offset_contribution_output_stage() local
725 const int batch_id = id.z() / depth_input; in run_offset_contribution_output_stage_symm() local
739 const int batch_id = id.z() / depth_input; in run_offset_contribution_output_stage_symm() local
H A DCpuGemmLowpOffsetContributionKernel.cpp138 const int batch_id = id.z() / depth_input; in run_offset_contribution() local
238 const int batch_id = id.z() / depth_input; in run_offset_contribution() local
297 const int batch_id = id.z() / depth_input; in run_offset_contribution() local
/aosp_15_r20/external/mesa3d/src/gallium/drivers/zink/
H A Dzink_screen.h42 zink_screen_update_last_finished(struct zink_screen *screen, uint64_t batch_id) in zink_screen_update_last_finished()
61 zink_screen_check_last_finished(struct zink_screen *screen, uint32_t batch_id) in zink_screen_check_last_finished()
H A Dzink_batch.c636 uint64_t batch_id = bs->fence.batch_id; in submit_queue() local
/aosp_15_r20/external/pytorch/aten/src/ATen/native/nested/cuda/
H A DNestedTensorTransformerFunctions.cu47 const int batch_id = blockIdx.x; in remove_padding_transform0213_2() local
88 const int batch_id = blockIdx.x; in remove_padding_2() local
121 const int batch_id = blockIdx.x; in remove_padding() local
257 const int batch_id = blockIdx.x; in add_padding_1() local
296 const int batch_id = blockIdx.x; in add_padding_2() local
341 const int batch_id = blockIdx.x; in add_padding_3() local
/aosp_15_r20/external/googleapis/google/cloud/visionai/v1/
H A Dlva_service.proto691 string batch_id = 4 [(google.api.field_behavior) = OUTPUT_ONLY]; field
697 string batch_id = 1; field
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/rnn/
H A Dlstm_ops_gpu.cu.cc90 const int batch_id = blockIdx.x * blockDim.x + threadIdx.x; in lstm_gates() local
311 const int batch_id = blockIdx.x * blockDim.x + threadIdx.x; in lstm_gates_bprop() local
/aosp_15_r20/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
H A Dtranspose.cc29 const std::string batch_id = in GetTransposeCode() local
H A Dstrided_slice.cc109 const std::string batch_id = in GetStridedSliceCode() local
/aosp_15_r20/external/cronet/net/third_party/quiche/src/quiche/quic/core/batch_writer/
H A Dquic_batch_writer_buffer.h42 uint32_t batch_id = 0; member
/aosp_15_r20/external/ComputeLibrary/src/core/CL/kernels/
H A DCLDepthToSpaceLayerKernel.cpp129 int batch_id = 0; in run() local
H A DCLSpaceToDepthLayerKernel.cpp128 int batch_id = 0; in run() local
H A DCLBatchToSpaceLayerKernel.cpp185 int batch_id = 0; in run() local
H A DCLSpaceToBatchLayerKernel.cpp207 int batch_id = 0; in run() local
/aosp_15_r20/external/cronet/net/quic/
H A Dquic_event_logger.cc35 uint32_t batch_id) { in NetLogQuicPacketSentParams()
501 uint32_t batch_id) { in OnPacketSent()
H A Dquic_connection_logger.cc168 uint32_t batch_id) { in OnPacketSent()
/aosp_15_r20/external/pytorch/aten/src/ATen/native/cuda/
H A DDilatedMaxPool2d.cu111 int batch_id = blockIdx.x % nbatch; in C10_LAUNCH_BOUNDS_1() local
225 int batch_id = blockIdx.x % nbatch; in C10_LAUNCH_BOUNDS_1() local
H A DAdaptiveAveragePooling.cu255 int batch_id = blockIdx.x % sizeB; in C10_LAUNCH_BOUNDS_1() local
372 int batch_id = blockIdx.x % sizeB; in C10_LAUNCH_BOUNDS_1() local
/aosp_15_r20/external/ComputeLibrary/src/core/NEON/kernels/
H A DNESpaceToDepthLayerKernel.cpp115 int batch_id = 0; in run() local
H A DNEBatchToSpaceLayerKernel.cpp164 int batch_id = 0; in run() local
H A DNESpaceToBatchLayerKernel.cpp173 int batch_id = 0; in run() local
/aosp_15_r20/external/googleapis/google/cloud/dataproc/v1/
H A Dbatches.proto95 string batch_id = 3 [(google.api.field_behavior) = OPTIONAL]; field
/aosp_15_r20/external/google-cloud-java/java-dataproc/proto-google-cloud-dataproc-v1/src/main/proto/google/cloud/dataproc/v1/
H A Dbatches.proto95 string batch_id = 3 [(google.api.field_behavior) = OPTIONAL]; field
/aosp_15_r20/external/tensorflow/tensorflow/core/kernels/
H A Dparameterized_truncated_normal_op_gpu.cu.cc93 const int64 batch_id = offset / samples_per_batch; in TruncatedNormalKernel() local
/aosp_15_r20/external/wpa_supplicant_8/src/eap_server/
H A Dtncs.c781 unsigned int batch_id; in tncs_process_if_tnccs() local

12