| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | kernels.cc | 97 for (int batch_idx = 0; batch_idx < batch_size; ++batch_idx) { in operator ()() local
|
| D | mat_mul_op.cc | 310 [&](int64_t batch_idx, int64_t row_begin, int64_t row_end) { in SparseDenseMatMulWithoutTransposedLHS()
381 [&](int64_t batch_idx, int64_t row_begin, int64_t row_end) { in SparseDenseMatMulWithTransposedLHS()
436 for (int64_t batch_idx = batch_begin; batch_idx <= batch_end_inclusive; in HandleBatchAndRowRange() local
|
| D | csr_sparse_matrix_to_dense_op.cc | 96 for (int64_t batch_idx = batch_begin; batch_idx < batch_end; in Compute() local
|
| D | csr_sparse_matrix_to_sparse_tensor_op.cc | 100 for (int64_t batch_idx = batch_begin; batch_idx < batch_end; in Compute() local
|
| D | sparse_mat_mul_op.cc | 187 for (int64_t batch_idx = batch_begin; batch_idx < batch_end; in Compute() local
229 for (int64_t batch_idx = batch_begin; batch_idx < batch_end; in Compute() local
|
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | unroll_batch_matmul.cc | 59 for (int batch_idx = 0; batch_idx < batch_size; ++batch_idx) { in SliceInput() local
212 for (int batch_idx = 0; batch_idx < bcast.output_batch_size(); ++batch_idx) { in Run() local
|
| /external/tensorflow/tensorflow/compiler/tf2xla/kernels/ |
| D | reverse_sequence_op.cc | 76 xla::XlaOp batch_idx = xla::Iota( in Compile() local
|
| /external/perfetto/src/trace_processor/containers/ |
| D | bit_vector_iterators.cc | 118 uint32_t batch_idx = set_bit_count_until_i++ % kBatchSize; in ReadSetBitBatch() local
|
| /external/ComputeLibrary/tests/validation/reference/ |
| D | QLSTMLayerNormalization.cpp | 60 for(uint32_t batch_idx = 0; batch_idx < num_batch; ++batch_idx) in qlstm_layer_normalization() local
|
| /external/tensorflow/tensorflow/core/kernels/ |
| D | gather_functor.h | 61 SliceIndex batch_idx = static_cast<SliceIndex>(start / indices_size); in HandleCopies() local
|
| D | gather_functor_batched.h | 63 SliceIndex batch_idx = static_cast<SliceIndex>( in HandleCopiesBatched() local
|
| D | count_ops.cc | 320 int batch_idx = 0; in Compute() local
|
| D | random_binomial_op.cc | 202 int64_t batch_idx = output_idx / samples_per_batch; in operator ()() local
|
| D | bincount_op.cc | 454 int batch_idx = 0; in Compute() local
|
| D | parameterized_truncated_normal_op.cc | 371 int64_t batch_idx = output_idx / samples_per_batch; in operator ()() local
|
| D | resource_variable_ops.cc | 761 for (int64_t batch_idx = 0, dest_idx = 0; batch_idx < batch_size; in AddBatchOffsets() local
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | non_max_suppression_op.cc | 307 void DoNMSPerClass(int batch_idx, int class_idx, const float* boxes_data, in DoNMSPerClass()
390 const int batch_idx, int total_size_per_batch, in SelectResultPerBatch()
476 int batch_idx = idx / num_classes; in BatchedNonMaxSuppressionOp() local
513 for (int batch_idx = begin; batch_idx < end; ++batch_idx) { in BatchedNonMaxSuppressionOp() local
556 int batch_idx = idx / per_batch_size; in BatchedNonMaxSuppressionOp() local
|
| /external/ComputeLibrary/tests/validation/fixtures/ |
| D | ROIAlignLayerFixture.h | 91 const auto batch_idx = dist_batch(gen); variable
|
| /external/ComputeLibrary/src/graph/mutators/ |
| D | GroupedConvolutionMutator.cpp | 57 …const unsigned int batch_idx = get_dimension_idx(weights_tensor_desc.layout, DataLay… in create_grouped_convolution() local
|
| /external/ComputeLibrary/src/core/NEON/kernels/ |
| D | NESpaceToBatchLayerKernel.cpp | 164 …const int batch_idx = get_data_layout_dimension_index(_data_layout, DataLayoutDimension::BATCHE… in run() local
|
| /external/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | unroll_batch_matmul.cc | 176 for (int batch_idx = 0; batch_idx < bcast.output_batch_size(); ++batch_idx) { in createMatMulOps() local
|
| /external/mesa3d/src/gallium/drivers/iris/ |
| D | iris_query.c | 68 int batch_idx; member
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.cc | 214 int batch_idx; member
|
| /external/ComputeLibrary/arm_compute/core/utils/misc/ |
| D | ShapeCalculator.h | 528 …const int batch_idx = get_data_layout_dimension_index(data_layout, DataLayoutDimension::B… in compute_deconvolution_output_shape() local
|