Searched refs:tensor_dim (Results 1 – 2 of 2) sorted by relevance
1892 tensor_dim = i - len(sizes.symbolic_sizes)1894 tensor_dim = i1912 actual_size = sizes.actual_sizes[tensor_dim]1919 tensor_dim, specified_size, actual_size,1931 'Tensor %s dimension' % tensor_name(sizes.x), tensor_dim,1941 size = sizes.actual_sizes[tensor_dim]1942 size_specifications[size_symbol] = (size, sizes.x, tensor_dim)
2347 const auto tensor_dim = tensor->getDimensions(); in ConvertConv2DHelper() local2348 const int c_dim_size = tensor_dim.d[params->use_implicit_batch ? 0 : 1]; in ConvertConv2DHelper()2381 input_dims = {static_cast<int>(tensor_dim.d[1]), in ConvertConv2DHelper()2382 static_cast<int>(tensor_dim.d[2])}; in ConvertConv2DHelper()3296 const auto tensor_dim = tensor->getDimensions(); in ConvertConv3DHelper() local3344 const int num_groups = (group == 0) ? tensor_dim.d[0] : group; in ConvertConv3DHelper()3608 const auto tensor_dim = tensor->getDimensions(); in ConvertFusedConv2DBiasActivation() local3612 input_dims = {static_cast<int>(tensor_dim.d[1]), in ConvertFusedConv2DBiasActivation()3613 static_cast<int>(tensor_dim.d[2])}; in ConvertFusedConv2DBiasActivation()3719 auto tensor_dim = tensor->getDimensions(); in ConvertPool() local[all …]