Home
last modified time | relevance | path

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

/external/ComputeLibrary/src/core/NEON/kernels/
DNEGEMMLowpOffsetContributionOutputStageKernel.cpp530 const int batch_id = id.z() / depth_input; in run_offset_contribution_output_stage() local
548 const int batch_id = id.z() / depth_input; in run_offset_contribution_output_stage() local
574 const int batch_id = id.z() / depth_input; in run_offset_contribution_output_stage() local
590 const int batch_id = id.z() / depth_input; in run_offset_contribution_output_stage() local
616 const int batch_id = id.z() / depth_input; in run_offset_contribution_output_stage() local
631 const int batch_id = id.z() / depth_input; in run_offset_contribution_output_stage() local
718 const int batch_id = id.z() / depth_input; in run_offset_contribution_output_stage_symm() local
732 const int batch_id = id.z() / depth_input; in run_offset_contribution_output_stage_symm() local
DNEGEMMLowpOffsetContributionKernel.cpp133 const int batch_id = id.z() / depth_input; in run_offset_contribution() local
232 const int batch_id = id.z() / depth_input; in run_offset_contribution() local
291 const int batch_id = id.z() / depth_input; in run_offset_contribution() local
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/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.cc108 const std::string batch_id = in GetStridedSliceCode() local
/external/ComputeLibrary/src/core/CL/kernels/
DCLDepthToSpaceLayerKernel.cpp128 int batch_id = 0; in run() local
DCLSpaceToDepthLayerKernel.cpp125 int batch_id = 0; in run() local
DCLBatchToSpaceLayerKernel.cpp184 int batch_id = 0; in run() local
DCLSpaceToBatchLayerKernel.cpp204 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/tensorflow/tensorflow/lite/delegates/gpu/common/task/
Dtensor_desc.cc557 std::string batch_id; in PerformGetWHOffsetSelector() local
/external/wpa_supplicant_8/src/eap_server/
Dtncs.c781 unsigned int batch_id; in tncs_process_if_tnccs() local
/external/wpa_supplicant_8/src/eap_peer/
Dtncc.c689 unsigned int batch_id; in tncc_process_if_tnccs() local
/external/tensorflow/tensorflow/core/kernels/batching_util/
Dshared_batch_scheduler_test.cc187 num_batches](int batch_id) -> int { in get_split_func()