| /external/tensorflow/tensorflow/core/kernels/ |
| D | spacetobatch_op.cc | 101 for (int block_dim = 0; block_dim < block_dims; ++block_dim) { in SpaceToBatchOpCompute() local
138 for (int block_dim = 0; block_dim < removed_prefix_block_dims; ++block_dim) { in SpaceToBatchOpCompute() local
146 for (int block_dim = removed_prefix_block_dims; in SpaceToBatchOpCompute() local
|
| 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
|
| D | spacetobatch_functor.cc | 107 for (int block_dim = 0; block_dim < NUM_BLOCK_DIMS; ++block_dim) { in operator ()() local
135 for (int block_dim = NUM_BLOCK_DIMS - 1; block_dim >= 0; --block_dim) { in operator ()() local
|
| 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
|
| D | spacetobatch_benchmark_test.cc | 32 for (int block_dim = 0; block_dim < num_block_dims; ++block_dim) { in ConstructSpaceToBatchGraph() local
|
| /external/tensorflow/tensorflow/core/profiler/protobuf/ |
| D | kernel_stats.proto | 16 repeated uint32 block_dim = 5; field
|
| /external/tensorflow/tensorflow/stream_executor/ |
| D | device_description.cc | 74 const BlockDim &block_dim = block_dim_limit(); in ToMap() local
|
| /external/tensorflow/tensorflow/lite/tools/optimize/sparsity/ |
| D | format_converter.cc | 62 int block_dim = 0; in FormatConverter() local
242 int block_dim = 0; in FormatConverter() local
|
| /external/tensorflow/tensorflow/tools/android/test/jni/object_tracking/ |
| D | flow_cache.h | 270 int block_dim = kNumCacheLevels; in BlockDimForCacheLevel() local
|
| /external/tensorflow/tensorflow/core/util/ |
| D | gpu_kernel_helper.h | 100 Status GpuLaunchKernel(void (*function)(Ts...), dim3 grid_dim, dim3 block_dim, in GpuLaunchKernel()
|
| /external/tensorflow/tensorflow/core/kernels/rnn/ |
| D | lstm_ops_gpu.cu.cc | 247 const int block_dim = 128; in LSTMBlockCellFpropWithCUDA() local
|
| /external/tensorflow/tensorflow/compiler/xla/service/ |
| D | triangular_solve_expander.cc | 153 int64 block_dim = (left_side) ? bdims - 2 : bdims - 1; in SolveWithInvertedDiagonalBlocks() local
|
| /external/tensorflow/tensorflow/core/kernels/image/ |
| D | non_max_suppression_op.cu.cc | 289 dim3 block_dim, thread_block; in NmsGpu() local
|
| /external/llvm-project/polly/lib/External/ppcg/ |
| D | gpu.h | 393 int block_dim[3]; member
|
| /external/vulkan-validation-layers/layers/ |
| D | buffer_validation.cpp | 4934 auto block_dim = FormatTexelBlockExtent(image_state->createInfo.format); in ValidateBufferBounds() local
|