Home
last modified time | relevance | path

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

/external/ComputeLibrary/src/cpu/kernels/
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
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
/external/mesa3d/src/gallium/drivers/zink/
Dzink_fence.h37 unsigned batch_id : 2; member
Dzink_batch.h43 unsigned batch_id : 2; member
Dzink_query.c40 unsigned batch_id : 2; //batch that the query was started in member
/external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/
Dtranspose.cc29 const std::string batch_id = in GetTransposeCode() local
Dstrided_slice.cc109 const std::string batch_id = in GetStridedSliceCode() local
/external/ComputeLibrary/src/core/CL/kernels/
DCLDepthToSpaceLayerKernel.cpp129 int batch_id = 0; in run() local
DCLSpaceToDepthLayerKernel.cpp128 int batch_id = 0; in run() local
DCLBatchToSpaceLayerKernel.cpp185 int batch_id = 0; in run() local
DCLSpaceToBatchLayerKernel.cpp207 int batch_id = 0; in run() local
/external/ComputeLibrary/src/core/NEON/kernels/
DNESpaceToDepthLayerKernel.cpp115 int batch_id = 0; in run() local
DNEBatchToSpaceLayerKernel.cpp164 int batch_id = 0; in run() local
DNESpaceToBatchLayerKernel.cpp173 int batch_id = 0; in run() local
/external/tensorflow/tensorflow/core/kernels/rnn/
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
/external/tensorflow/tensorflow/core/kernels/
Dparameterized_truncated_normal_op_gpu.cu.cc93 const int64 batch_id = offset / samples_per_batch; in TruncatedNormalKernel() local
/external/wpa_supplicant_8/src/eap_peer/
Dtncc.c689 unsigned int batch_id; in tncc_process_if_tnccs() local
/external/wpa_supplicant_8/src/eap_server/
Dtncs.c781 unsigned int batch_id; in tncs_process_if_tnccs() local
/external/tensorflow/tensorflow/python/distribute/
Dinput_lib_test.py1914 def data_fn(batch_id) -> OuterType: argument