| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/ |
| H A D | spacetobatch_op.cc | 101 for (int block_dim = 0; block_dim < block_dims; ++block_dim) { in SpaceToBatchOpCompute() local
150 for (int block_dim = 0; block_dim < removed_prefix_block_dims; ++block_dim) { in SpaceToBatchOpCompute() local
158 for (int block_dim = removed_prefix_block_dims; in SpaceToBatchOpCompute() local
|
| H A D | batchtospace_op.cc | 98 for (int block_dim = 0; block_dim < block_dims; ++block_dim) { in BatchToSpaceOpCompute() local
140 for (int block_dim = 0; block_dim < removed_prefix_block_dims; ++block_dim) { in BatchToSpaceOpCompute() local
148 for (int block_dim = removed_prefix_block_dims; in BatchToSpaceOpCompute() local
|
| H A D | spacetobatch_functor.cc | 109 for (int block_dim = 0; block_dim < NUM_BLOCK_DIMS; ++block_dim) { in operator ()() local
137 for (int block_dim = NUM_BLOCK_DIMS - 1; block_dim >= 0; --block_dim) { in operator ()() local
|
| H A D | spacetobatch_functor_gpu.cu.cc | 67 for (int block_dim = NUM_BLOCK_DIMS - 1; block_dim >= 0; --block_dim) { in S2B() local
117 for (int block_dim = 0; block_dim < NUM_BLOCK_DIMS; ++block_dim) { in operator ()() local
|
| H A D | spacetobatch_benchmark_test.cc | 32 for (int block_dim = 0; block_dim < num_block_dims; ++block_dim) { in ConstructSpaceToBatchGraph() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/lite/kernels/internal/utils/ |
| H A D | sparsity_format_converter.cc | 64 int block_dim = 0; in FormatConverter() local
280 int block_dim = 0; in InitSparseToDenseConverter() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/profiler/protobuf/ |
| H A D | kernel_stats.proto | 16 repeated uint32 block_dim = 5; field
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/stream_executor/ |
| H A D | device_description.cc | 70 const BlockDim &block_dim = block_dim_limit(); in ToMap() local
|
| /aosp_15_r20/external/pytorch/aten/src/ATen/ |
| H A D | SparseCsrTensorImpl.h | 82 inline int64_t block_dim() const noexcept { in block_dim() function
|
| /aosp_15_r20/external/tensorflow/tensorflow/tools/android/test/jni/object_tracking/ |
| H A D | flow_cache.h | 270 int block_dim = kNumCacheLevels; in BlockDimForCacheLevel() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/util/ |
| H A D | gpu_kernel_helper.h | 102 Status GpuLaunchKernel(void (*function)(Ts...), dim3 grid_dim, dim3 block_dim, in GpuLaunchKernel()
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/rnn/ |
| H A D | lstm_ops_gpu.cu.cc | 247 const int block_dim = 128; in LSTMBlockCellFpropWithCUDA() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/compiler/xla/service/ |
| H A D | triangular_solve_expander.cc | 147 int64_t block_dim = (left_side) ? bdims - 2 : bdims - 1; in SolveWithInvertedDiagonalBlocks() local
|
| /aosp_15_r20/external/tensorflow/tensorflow/core/kernels/image/ |
| H A D | non_max_suppression_op.cu.cc | 707 dim3 block_dim, thread_block; in NmsGpu() local
|
| /aosp_15_r20/external/libaom/av1/common/ |
| H A D | av1_loopfilter.c | 719 const int block_dim = in set_one_param_for_line_luma() local
|
| /aosp_15_r20/external/vulkan-validation-layers/layers/ |
| H A D | buffer_validation.cpp | 4934 auto block_dim = FormatTexelBlockExtent(image_state->createInfo.format); in ValidateBufferBounds() local
|