| /external/XNNPACK/src/ |
| D | operator-run.c | 73 size_t batch_index, in xnn_compute_spmm()
90 size_t batch_index, in xnn_compute_gigemm()
117 size_t batch_index, in xnn_compute_igemm()
143 size_t batch_index, in xnn_compute_gsubconv2d()
182 size_t batch_index, in xnn_compute_subconv2d()
220 size_t batch_index, in xnn_compute_dconv2d_hwc2spchw()
257 size_t batch_index, in xnn_compute_dwconv2d_spchw()
275 size_t batch_index, in xnn_compute_argmax_pooling_unipass()
295 size_t batch_index, in xnn_compute_argmax_pooling_multipass()
318 size_t batch_index, in xnn_compute_max_pooling()
[all …]
|
| D | indirection.c | 103 for (size_t batch_index = batch_start; batch_index < batch_size; batch_index++) { in xnn_indirection_init_dwconv2d() local
|
| /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/kernels/ |
| D | sparse_cross_op.cc | 148 void Update(const int64 batch_index, const int64 cross_count, in Update()
175 string Generate(const int64 batch_index, in Generate()
200 int64 Generate(const int64 batch_index, in Generate()
230 int64 batch_index) in ProductIterator()
540 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 | 1104 auto batch_index = in OutputBatch() local
1156 const auto batch_index = in Compute() local
|
| /external/tensorflow/tensorflow/core/util/ |
| D | padding.cc | 59 const int32 batch_index = GetTensorBatchDimIndex(num_dims, data_format); in CheckValidPadding() local
|
| D | example_proto_helper.cc | 217 const Example& example, const string& example_name, const int batch_index, in SingleExampleProtoToTensors()
|
| /external/tensorflow/tensorflow/python/keras/engine/ |
| D | training_v2_utils.py | 209 def _predict_on_batch(x, y=None, sample_weights=None, batch_index=None): argument
276 def _gather_result_by_index(input_tensor, batch_index): argument
|
| /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 | 237 for (size_t batch_index = 0; batch_index < batch_size(); batch_index++) { in TestF32() 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 | 963 llvm::Value* batch_index, in SliceOutInnerArray()
|