| /external/tensorflow/tensorflow/core/kernels/sparse/ |
| D | kernels.cc | 91 for (int batch_idx = 0; batch_idx < batch_size; ++batch_idx) { in operator ()() local
|
| D | csr_sparse_matrix_to_dense_op.cc | 96 for (int64 batch_idx = batch_begin; batch_idx < batch_end; ++batch_idx) { in Compute() local
|
| D | csr_sparse_matrix_to_sparse_tensor_op.cc | 100 for (int64 batch_idx = batch_begin; batch_idx < batch_end; ++batch_idx) { in Compute() local
|
| D | mat_mul_op.cc | 307 [&](int64 batch_idx, int64 row_begin, int64 row_end) { in SparseDenseMatMulWithoutTransposedLHS()
378 [&](int64 batch_idx, int64 row_begin, int64 row_end) { in SparseDenseMatMulWithTransposedLHS()
433 for (int64 batch_idx = batch_begin; batch_idx <= batch_end_inclusive; in HandleBatchAndRowRange() local
|
| D | sparse_mat_mul_op.cc | 187 for (int64 batch_idx = batch_begin; batch_idx < batch_end; in Compute() local
229 for (int64 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/tensorflow/tensorflow/compiler/mlir/tensorflow/transforms/ |
| D | unroll_batch_matmul.cc | 100 for (int batch_idx = 0; batch_idx < batch_size; ++batch_idx) { in sliceInput() local
162 for (int batch_idx = 0; batch_idx < bcast.output_batch_size(); ++batch_idx) { in createMatMulOps() 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 | 287 int batch_idx = 0; in Compute() local
|
| D | random_binomial_op.cc | 202 int64 batch_idx = output_idx / samples_per_batch; in operator ()() local
|
| D | bincount_op.cc | 421 int batch_idx = 0; in Compute() local
|
| D | parameterized_truncated_normal_op.cc | 371 int64 batch_idx = output_idx / samples_per_batch; in operator ()() local
|
| D | resource_variable_ops.cc | 722 for (int64 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 | 304 void DoNMSPerClass(int batch_idx, int class_idx, const float* boxes_data, in DoNMSPerClass()
387 const int batch_idx, int total_size_per_batch, in SelectResultPerBatch()
473 int batch_idx = idx / num_classes; in BatchedNonMaxSuppressionOp() local
510 for (int batch_idx = begin; batch_idx < end; ++batch_idx) { in BatchedNonMaxSuppressionOp() local
553 int batch_idx = idx / per_batch_size; in BatchedNonMaxSuppressionOp() local
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | dnn.cc | 195 int depth_idx, batch_idx, spatial_idx; in GetDimIndices() local
|
| /external/mesa3d/src/gallium/drivers/iris/ |
| D | iris_query.c | 68 int batch_idx; member
|