| /external/XNNPACK/src/ |
| D | operator-run.c | 346 size_t batch_index, in xnn_compute_spmm()
364 size_t batch_index, in xnn_compute_grouped_batch_igemm()
417 size_t batch_index, in xnn_compute_batch_igemm()
468 size_t batch_index, in xnn_compute_grouped_subgemm2d()
506 size_t batch_index, in xnn_compute_subgemm2d()
543 size_t batch_index, in xnn_compute_grouped_subconv2d()
582 size_t batch_index, in xnn_compute_subconv2d()
620 size_t batch_index, in xnn_compute_conv2d_hwc2chw()
642 size_t batch_index, in xnn_compute_dwconv_unipass()
661 size_t batch_index, in xnn_compute_dwconv2d_chw()
[all …]
|
| /external/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLCrop.cpp | 54 …sor *src, ICLTensor *dst, Coordinates2D start, Coordinates2D end, uint32_t batch_index, float extr… in configure()
60 …sor *src, ICLTensor *dst, Coordinates2D start, Coordinates2D end, uint32_t batch_index, float extr… in configure()
73 …onst ITensorInfo *output, Coordinates2D start, Coordinates2D end, uint32_t batch_index, float extr… in validate()
|
| D | CLCropResize.cpp | 40 …Tensor *output, uint32_t crop_box_ind, Coordinates &start, Coordinates &end, uint32_t &batch_index) in configure_crop()
140 uint32_t batch_index; in configure() local
|
| /external/ComputeLibrary/src/gpu/cl/operators/ |
| D | ClCrop.cpp | 35 …o *src, ITensorInfo *dst, Coordinates2D start, Coordinates2D end, uint32_t batch_index, float extr… in configure()
44 …, const ITensorInfo *dst, Coordinates2D start, Coordinates2D end, uint32_t batch_index, float extr… in validate()
|
| /external/ComputeLibrary/src/gpu/cl/kernels/ |
| D | ClCropKernel.cpp | 49 …t ITensorInfo *src, ITensorInfo *dst, Coordinates2D start, Coordinates2D end, uint32_t batch_index, in configure()
90 …, const ITensorInfo *dst, Coordinates2D start, Coordinates2D end, uint32_t batch_index, float extr… in validate()
|
| /external/XNNPACK/test/ |
| D | resize-bilinear-operator-tester.h | 244 for (size_t batch_index = 0; batch_index < batch_size(); batch_index++) { in TestNHWCxF16() local
329 for (size_t batch_index = 0; batch_index < batch_size(); batch_index++) { in TestNHWCxF32() local
411 for (size_t batch_index = 0; batch_index < batch_size(); batch_index++) { in TestNHWCxS8() local
494 for (size_t batch_index = 0; batch_index < batch_size(); batch_index++) { in TestNHWCxU8() local
580 for (size_t batch_index = 0; batch_index < batch_size(); batch_index++) { in TestNCHWxF32() local
|
| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | sparse_cholesky_op.cc | 115 for (int64_t batch_index = batch_begin; batch_index < batch_end; in Compute() local
194 for (int64_t batch_index = batch_begin; batch_index < batch_end; in Compute() local
|
| D | sparse_ordering_amd_op.cc | 100 for (int64_t batch_index = batch_begin; batch_index < batch_end; in Compute() local
|
| D | sparse_mat_mul_op.cc | 266 const CSRSparseMatrix& csr_matrix, const int rank, const int batch_index, in GetSparseMatrixRef()
|
| D | mat_mul_op.cc | 270 const CSRSparseMatrix& csr_matrix, const int batch_index, in GetSparseMatrixRef()
|
| /external/tensorflow/tensorflow/core/util/ |
| D | padding.cc | 52 const int32_t batch_index = GetTensorBatchDimIndex(num_dims, data_format); in CheckValidPadding() local
|
| D | example_proto_helper.cc | 219 const Example& example, const string& example_name, const int batch_index, in SingleExampleProtoToTensors()
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | ragged_cross_op.cc | 205 void WriteCombination(int64_t batch_index, in WriteCombination()
217 void WriteCombination(int64_t batch_index, in WriteCombination()
236 void NextCombination(int64_t batch_index, in NextCombination()
612 int batch_index) { in CrossCountByBatchIndex()
|
| D | sparse_cross_op.cc | 283 void Update(const int64_t batch_index, const int64_t cross_count, in Update()
312 string Generate(const int64_t batch_index, in Generate()
338 int64_t Generate(const int64_t batch_index, in Generate()
371 int64_t Generate(const int64_t batch_index, in Generate()
403 int64_t batch_index) in ProductIterator()
624 int batch_index) { in CrossCountByBatchIndex()
|
| D | in_topk_op_gpu.cu.cc | 45 const int batch_index = i / num_classes; in ComputePredictionMaskKernel() local
|
| D | batch_kernels.cc | 860 auto batch_index = in OutputBatch() local
922 const auto batch_index = in Compute() local
|
| D | topk_op_gpu.h | 347 const int batch_index = blockIdx.x; in TopKKernel() local
|
| /external/mesa3d/src/mesa/state_tracker/ |
| D | st_cb_perfmon.h | 34 unsigned batch_index; member
|
| /external/tensorflow/tensorflow/python/keras/layers/preprocessing/benchmarks/ |
| D | feature_column_benchmark.py | 56 def on_predict_batch_begin(self, batch_index, _): argument
|
| /external/ComputeLibrary/tests/validation/reference/ |
| D | CropResize.cpp | 124 …ge(const SimpleTensor<T> &src, Coordinates start, Coordinates end, int32_t batch_index, float extr… in crop_image()
|
| /external/ComputeLibrary/src/core/NEON/kernels/ |
| D | NECropKernel.cpp | 303 …uint32_t batch_index = *(reinterpret_cast<int32_t *>(_box_ind->ptr_to_element(Coordinates(_crop… in run() local
|
| /external/mesa3d/src/intel/tools/ |
| D | intel_dump_gpu.c | 216 int batch_index; in dump_execbuffer2() local
|
| /external/igt-gpu-tools/tools/ |
| D | aubdump.c | 909 int batch_index; in dump_execbuffer2() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/cpu/ |
| D | dot_op_emitter.cc | 1305 llvm::Value* batch_index, in SliceOutInnerArray()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | space_to_batch_converter.cc | 2214 const int64_t batch_index = (k / (new_space_size + halo_size)); in Propagate() local
2325 const int64_t batch_index = (radix[1] / batch_residue) % num_splits; in SelectValidPortion() local
|