Searched defs:time_major (Results 1 - 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/contrib/lite/kernels/ |
H A D | unidirectional_sequence_rnn.cc | 56 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 D | unidirectional_sequence_rnn_test.cc | 128 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 D | builtin_op_data.h | 87 bool time_major; member in struct:__anon25871
|
/external/tensorflow/tensorflow/contrib/lite/schema/ |
H A D | schema_generated.h | 1907 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