Searched refs:batch_id (Results 1 – 14 of 14) sorted by relevance
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/tasks/ |
| D | transpose.cc | 29 const std::string batch_id = in GetTransposeCode() local
58 const std::string bhw[] = {batch_id, "Y", "X"}; in GetTransposeCode()
73 const std::string bhwc[] = {batch_id, "Y", "X", "dst_channel"}; in GetTransposeCode()
|
| D | strided_slice.cc | 108 const std::string batch_id = in GetStridedSliceCode() local
129 c += " int s_b = " + batch_id + " * args.stride_b + args.offset_b;\n"; in GetStridedSliceCode()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | parameterized_truncated_normal_op_gpu.cu.cc | 93 const int64 batch_id = offset / samples_per_batch; in TruncatedNormalKernel() local
94 T mean = means[single_mean ? 0 : batch_id]; in TruncatedNormalKernel()
95 const T input_stddev = stddevs[single_stddev ? 0 : batch_id]; in TruncatedNormalKernel()
96 T minval = minvals[single_minval ? 0 : batch_id]; in TruncatedNormalKernel()
97 T maxval = maxvals[single_maxval ? 0 : batch_id]; in TruncatedNormalKernel()
|
| /external/mesa3d/src/gallium/drivers/zink/ |
| D | zink_fence.c | 64 ret->batch_id = batch->batch_id; in zink_create_fence()
106 p_atomic_set(&res->batch_uses[fence->batch_id], 0); in fence_remove_resource_access()
|
| D | zink_batch.c | 134 res->batch_uses[batch->batch_id] |= mask; in zink_batch_reference_resource_rw()
137 stencil->batch_uses[batch->batch_id] |= mask; in zink_batch_reference_resource_rw()
|
| D | zink_query.c | 40 unsigned batch_id : 2; //batch that the query was started in member
357 q->batch_id = batch->batch_id; in begin_query()
386 q->batch_id = batch->batch_id; in end_query()
|
| D | zink_fence.h | 37 unsigned batch_id : 2; member
|
| D | zink_batch.h | 43 unsigned batch_id : 2; member
|
| D | zink_context.c | 1318 ctx->batches[i].batch_id = i; in zink_context_create()
|
| /external/tensorflow/tensorflow/core/kernels/rnn/ |
| D | lstm_ops_gpu.cu.cc | 90 const int batch_id = blockIdx.x * blockDim.x + threadIdx.x; in lstm_gates() local
96 if (batch_id >= batch_size || act_id >= cell_size) return; in lstm_gates()
142 const int gid = batch_id * cell_size * 4 + act_id; in lstm_gates()
143 const int cid = batch_id * cell_size + act_id; in lstm_gates()
311 const int batch_id = blockIdx.x * blockDim.x + threadIdx.x; in lstm_gates_bprop() local
314 if (batch_id >= batch_size || act_id >= cell_size) return; in lstm_gates_bprop()
316 const int gid = batch_id * cell_size * 4 + act_id; in lstm_gates_bprop()
317 const int cid = batch_id * cell_size + act_id; in lstm_gates_bprop()
|
| /external/tensorflow/tensorflow/python/keras/ |
| D | callbacks_v1.py | 265 self.batch_id = batch_id = array_ops.placeholder(dtypes.int32)
278 batch = state_ops.assign(embedding[batch_id:batch_id + step],
453 feed_dict.update({self.batch_id: i, self.step: step})
|
| /external/wpa_supplicant_8/src/eap_server/ |
| D | tncs.c | 781 unsigned int batch_id; in tncs_process_if_tnccs() local
810 batch_id = atoi(pos); in tncs_process_if_tnccs()
812 batch_id); in tncs_process_if_tnccs()
813 if (batch_id != tncs->last_batchid + 1) { in tncs_process_if_tnccs()
816 batch_id, tncs->last_batchid + 1); in tncs_process_if_tnccs()
820 tncs->last_batchid = batch_id; in tncs_process_if_tnccs()
|
| /external/wpa_supplicant_8/src/eap_peer/ |
| D | tncc.c | 689 unsigned int batch_id; in tncc_process_if_tnccs() local
722 batch_id = atoi(pos); in tncc_process_if_tnccs()
724 batch_id); in tncc_process_if_tnccs()
725 if (batch_id != tncc->last_batchid + 1) { in tncc_process_if_tnccs()
728 batch_id, tncc->last_batchid + 1); in tncc_process_if_tnccs()
732 tncc->last_batchid = batch_id; in tncc_process_if_tnccs()
|
| /external/tensorflow/tensorflow/lite/delegates/gpu/common/task/ |
| D | tensor_desc.cc | 518 std::string batch_id; in PerformGetWHOffsetSelector() local
523 batch_id = it->second; in PerformGetWHOffsetSelector()
526 ")) * batch + (", batch_id, ")"); in PerformGetWHOffsetSelector()
|