Searched defs:output_size_ (Results 1 – 12 of 12) sorted by relevance
428 auto output_size_ = in adaptive_max_pool2d_with_indices() local484 auto output_size_ = in adaptive_max_pool3d_with_indices() local571 auto output_size_ = in adaptive_avg_pool2d() local602 auto output_size_ = in adaptive_avg_pool3d() local647 std::vector<int64_t> output_size_; in _unpool_output_size() local692 auto output_size_ = in max_unpool1d() local737 auto output_size_ = in max_unpool2d() local780 auto output_size_ = in max_unpool3d() local829 std::optional<ExpandingArray<2>> output_size_ = output_size; in fractional_max_pool2d_with_indices() local923 std::optional<ExpandingArray<3>> output_size_ = output_size; in fractional_max_pool3d_with_indices() local
58 const int output_size_; variable
60 const int output_size_; variable
80 int32_t output_size_ = 1; variable
78 int output_size_; member in tflite::__anon28c702c10111::LSHProjectionOpModel
157 int output_size_; member in tflite::__anonc69a0ae70111::QuantizedLSTMOpModel
316 std::vector<int32_t> output_size_ = {n_batch_, n_output_}; member in tflite::__anondd2239e50111::BaseLstmParam
177 std::optional<at::IntArrayRef> output_size_ = output_size; in _output_padding() local
174 auto output_size_ = output_size.value(); in NestedTensor_to_padded_tensor_cuda() local
229 int64_t output_size_; member in tensorflow::__anon6d95e0f40111::TestGpuSelectFlaggedKernel
314 auto output_size_ = output_size.value(); in NestedTensor_to_padded_tensor_generic() local
2144 int output_size_; member in tflite::__anon8716bded0111::LSHProjectionOpModel