Searched refs:LogSoftmaxOptions (Results 1 – 12 of 12) sorted by relevance
16 LogSoftmaxOptions::LogSoftmaxOptions(int64_t dim) : dim_(dim) {} in LogSoftmaxOptions() function in torch::nn::LogSoftmaxOptions
53 "builtin_options_type": "LogSoftmaxOptions", in LogSoftmaxFixture()
309 struct TORCH_API LogSoftmaxOptions { struct310 LogSoftmaxOptions(int64_t dim);
297 : LogSoftmaxImpl(LogSoftmaxOptions(dim)) {} in LogSoftmaxImpl()298 explicit LogSoftmaxImpl(const LogSoftmaxOptions& options_);307 LogSoftmaxOptions options;
152 LogSoftmaxImpl::LogSoftmaxImpl(const LogSoftmaxOptions& options_) in LogSoftmaxImpl()
395 LogSoftmaxOptions,769 table LogSoftmaxOptions {
421 LogSoftmaxOptions,822 table LogSoftmaxOptions {
456 LogSoftmaxOptions,875 table LogSoftmaxOptions {
237 struct LogSoftmaxOptions;1947 template<> struct BuiltinOptionsTraits<tflite::LogSoftmaxOptions> {7927 typedef LogSoftmaxOptions TableType;7930 struct LogSoftmaxOptions FLATBUFFERS_FINAL_CLASS : private flatbuffers::Table {7939 …static flatbuffers::Offset<LogSoftmaxOptions> Pack(flatbuffers::FlatBufferBuilder &_fbb, const Log…7943 typedef LogSoftmaxOptions Table;7950 flatbuffers::Offset<LogSoftmaxOptions> Finish() {7952 auto o = flatbuffers::Offset<LogSoftmaxOptions>(end);7957 inline flatbuffers::Offset<LogSoftmaxOptions> CreateLogSoftmaxOptions(7963 flatbuffers::Offset<LogSoftmaxOptions> CreateLogSoftmaxOptions(flatbuffers::FlatBufferBuilder &_fbb…[all …]
390 LogSoftmaxOptions,759 table LogSoftmaxOptions {
513 LogSoftmaxOptions,1132 table LogSoftmaxOptions {
5326 c10::str(LogSoftmax(LogSoftmaxOptions(1))), in TEST_F()