| /external/XNNPACK/src/ |
| D | operator-run.c | 74 size_t batch_index, in xnn_compute_spmm()
92 size_t batch_index, in xnn_compute_grouped_batch_igemm()
145 size_t batch_index, in xnn_compute_batch_igemm()
196 size_t batch_index, in xnn_compute_grouped_subgemm2d()
234 size_t batch_index, in xnn_compute_subgemm2d()
271 size_t batch_index, in xnn_compute_grouped_subconv2d()
310 size_t batch_index, in xnn_compute_subconv2d()
348 size_t batch_index, in xnn_compute_conv2d_hwc2chw()
370 size_t batch_index, in xnn_compute_dwconv_unipass()
389 size_t batch_index, in xnn_compute_dwconv2d_chw()
[all …]
|
| /external/ComputeLibrary/src/core/CL/kernels/ |
| D | CLCropKernel.cpp | 45 …input, ICLTensor *output, Coordinates2D start, Coordinates2D end, uint32_t batch_index, float extr… in configure()
50 … ICLTensor *input, ICLTensor *output, Coordinates2D start, Coordinates2D end, uint32_t batch_index, in configure()
93 …onst ITensorInfo *output, Coordinates2D start, Coordinates2D end, uint32_t batch_index, float extr… in validate()
|
| /external/XNNPACK/test/ |
| D | resize-bilinear-operator-tester.h | 241 for (size_t batch_index = 0; batch_index < batch_size(); batch_index++) { in TestNHWCxF32() local
323 for (size_t batch_index = 0; batch_index < batch_size(); batch_index++) { in TestNHWCxS8() local
406 for (size_t batch_index = 0; batch_index < batch_size(); batch_index++) { in TestNHWCxU8() local
492 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 | 264 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/ComputeLibrary/src/runtime/CL/functions/ |
| D | CLCropResize.cpp | 43 …Tensor *output, uint32_t crop_box_ind, Coordinates &start, Coordinates &end, uint32_t &batch_index) in configure_crop()
142 uint32_t batch_index; in configure() local
|
| /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 | 278 void Update(const int64_t batch_index, const int64_t cross_count, in Update()
307 string Generate(const int64_t batch_index, in Generate()
333 int64 Generate(const int64_t batch_index, const std::vector<int>& permutation, in Generate()
365 int64 Generate(const int64_t batch_index, const std::vector<int>& permutation, in Generate()
396 int64_t batch_index) in ProductIterator()
617 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 | 846 auto batch_index = in OutputBatch() local
898 const auto batch_index = in Compute() local
|
| D | topk_op_gpu.h | 346 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 | 373 …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 | 1126 llvm::Value* batch_index, in SliceOutInnerArray()
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | space_to_batch_converter.cc | 1930 const int64_t batch_index = (k / (new_space_size + halo_size)); in Propagate() local
2024 const int64_t batch_index = (k / new_space_size) % num_splits; in SelectValidPortion() local
|