Searched defs:batch_data (Results 1 – 5 of 5) sorted by relevance
328 BatchData* batch_data = CreateBatch(2, false /* set_on_complete */); in AddBatchForInternalRecvTrailingMetadata() local461 BatchData* batch_data = in AddBatchesForPendingBatches() local791 RefCountedPtr<BatchData> batch_data(static_cast<BatchData*>(arg)); in RecvInitialMetadataReady() local885 RefCountedPtr<BatchData> batch_data(static_cast<BatchData*>(arg)); in RecvMessageReady() local1075 RefCountedPtr<BatchData> batch_data(static_cast<BatchData*>(arg)); in RecvTrailingMetadataReady() local1229 RefCountedPtr<BatchData> batch_data(static_cast<BatchData*>(arg)); in OnComplete() local1301 RefCountedPtr<BatchData> batch_data(static_cast<BatchData*>(arg)); in OnCompleteForCancelOp() local
60 const float *batch_data = in_ptr + param->in_strides_[kNHWC_N] * roi_batch_ind; in ROIPooling() local
170 auto batch_data = tensor_data + batch_index * tensor.ElementsNum(); in AddBatch() local
2875 uint32_t batch_data[512]; member2903 uint32_t batch_data[9]; member
4938 uint32_t batch_data[480]; member5023 uint32_t batch_data[9]; member