Searched defs:time_major (Results 1 - 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/kernels/
H A Dunidirectional_sequence_rnn.cc56 const bool time_major = params->time_major; local
58 (time_major) ? input->dims->data[1] : input->dims->data[0];
60 (time_major) ? input->dims->data[0] : input->dims->data[1];
83 output_size_array->data[0] = (time_major) ? max_time : batch_size;
84 output_size_array->data[1] = (time_major) ? batch_size : max_time;
108 const bool time_major = params->time_major; local
110 (time_major) ? input->dims->data[1] : input->dims->data[0];
112 (time_major)
[all...]
H A Dunidirectional_sequence_rnn_test.cc128 bool time_major)
141 CreateSequenceRNNOptions(builder_, time_major,
144 if (time_major) {
208 /*units=*/16, /*size=*/8, /*time_major=*/false);
275 /*units=*/16, /*size=*/8, /*time_major=*/true);
127 UnidirectionalRNNOpModel(int batches, int sequence_len, int units, int size, bool time_major) argument
/external/tensorflow/tensorflow/contrib/lite/
H A Dbuiltin_op_data.h87 bool time_major; member in struct:__anon25871
/external/tensorflow/tensorflow/contrib/lite/schema/
H A Dschema_generated.h1907 bool time_major; member in struct:tflite::SequenceRNNOptionsT
1910 : time_major(false),
1921 bool time_major() const { function in struct:tflite::FLATBUFFERS_FINAL_CLASS
1941 void add_time_major(bool time_major) { argument
1942 fbb_.AddElement<uint8_t>(SequenceRNNOptions::VT_TIME_MAJOR, static_cast<uint8_t>(time_major), 0);
1961 bool time_major = false,
1965 builder_.add_time_major(time_major);
1973 bool time_major; member in struct:tflite::BidirectionalSequenceRNNOptionsT
1976 : time_major(false),
1987 bool time_major() cons function in struct:tflite::FLATBUFFERS_FINAL_CLASS
2007 add_time_major(bool time_major) argument
[all...]

Completed in 1478 milliseconds