Searched refs:kMinDimensionToUseTiles (Results 1 - 1 of 1) sorted by relevance

/external/tensorflow/tensorflow/core/kernels/
H A Dconv_ops_gpu_3.cu.cc814 T* output, const int kMinDimensionToUseTiles) {
856 int requested_tile_size_i = input_dims[1] >= kMinDimensionToUseTiles
859 int requested_tile_size_j = input_dims[1] >= kMinDimensionToUseTiles
901 static const int kMinDimensionToUseTiles = 16; local
904 bool large_matrix = input_dims[1] >= kMinDimensionToUseTiles &&
905 input_dims[2] >= kMinDimensionToUseTiles;
930 d, input, input_dims, output, kMinDimensionToUseTiles);
812 SwapDimension1And2InTensor3WithNarrowMatrices( const GPUDevice& d, const T* input, const Dimension<3>& input_dims, T* output, const int kMinDimensionToUseTiles) argument

Completed in 118 milliseconds