| /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/tensorflow/tensorflow/core/kernels/sparse/ |
| D | sparse_cholesky_op.cc | 113 for (int64 batch_index = batch_begin; batch_index < batch_end; in Compute() local
192 for (int64 batch_index = batch_begin; batch_index < batch_end; in Compute() local
|
| D | sparse_ordering_amd_op.cc | 100 for (int64 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 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 | 204 void WriteCombination(int64 batch_index, const std::vector<int>& combination, in WriteCombination()
216 void WriteCombination(int64 batch_index, const std::vector<int>& combination, in WriteCombination()
235 void NextCombination(int64 batch_index, std::vector<int>* combination) const { in NextCombination()
580 int batch_index) { in CrossCountByBatchIndex()
|
| D | sparse_cross_op.cc | 269 void Update(const int64 batch_index, const int64 cross_count, in Update()
297 string Generate(const int64 batch_index, const std::vector<int>& permutation, in Generate()
322 int64 Generate(const int64 batch_index, const std::vector<int>& permutation, in Generate()
354 int64 Generate(const int64 batch_index, const std::vector<int>& permutation, in Generate()
385 int64 batch_index) in ProductIterator()
576 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 | 773 auto batch_index = in OutputBatch() local
825 const auto batch_index = in Compute() local
|
| /external/mesa3d/src/mesa/state_tracker/ |
| D | st_cb_perfmon.h | 34 unsigned batch_index; member
|
| /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
326 for (size_t batch_index = 0; batch_index < batch_size(); batch_index++) { in TestNCHWxF32() local
|
| /external/mesa3d/src/intel/tools/ |
| D | intel_dump_gpu.c | 216 int batch_index; in dump_execbuffer2() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | space_to_batch_converter.cc | 1577 const int64 batch_index = (k / (new_space_size + halo_size)); in Propagate() local
1671 const int64 batch_index = (k / new_space_size) % num_splits; in SelectValidPortion() 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 | 1107 llvm::Value* batch_index, in SliceOutInnerArray()
|