Home
last modified time | relevance | path

Searched refs:max_dim (Results 1 – 3 of 3) sorted by relevance

/external/tensorflow/tensorflow/compiler/tf2tensorrt/convert/
Dtrt_optimization_pass.cc201 int max_dim = -1; in Optimize() local
206 if (shape.dim_size(0) > max_dim) max_dim = shape.dim_size(0); in Optimize()
211 if (max_dim > 0) { in Optimize()
212 maximum_batch_size_ = max_dim; in Optimize()
213 VLOG(1) << "Setting maximum batch size to " << max_dim; in Optimize()
222 if (max_dim > maximum_batch_size_) { in Optimize()
224 << " is less than input batch size " << max_dim in Optimize()
/external/tensorflow/tensorflow/core/kernels/
Deigen_mkldnn_contraction_kernel_test.cc24 Eigen::array<Index, NumDims> RandomDims(int min_dim = 1, int max_dim = 20) { in RandomDims() argument
27 dims[i] = internal::random<int>(min_dim, max_dim); in RandomDims()
/external/mesa3d/src/gallium/drivers/svga/
Dsvga_resource_texture.c925 unsigned max_dim = template->width0; in svga_texture_create() local
936 max_dim = MAX2(max_dim, template->height0); in svga_texture_create()
939 max_dim = MAX3(max_dim, template->height0, template->depth0); in svga_texture_create()
949 assert(1 << template->last_level <= max_dim); in svga_texture_create()