HomeSort by relevance Sort by last modified time
    Searched full:dim2_size (Results 1 - 3 of 3) sorted by null

  /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
lstm_utils.cc 56 int dim2_size, const Array& original_array,
59 bool is_bias = dim2_size == 1;
60 Shape shape = is_bias ? Shape({dim1_size}) : Shape({dim1_size, dim2_size});
67 start_idx1, start_idx2, buffer, dim2_size, 0, 0, dim1_size,
68 dim2_size);
lstm_utils.h 89 int dim2_size, const Array& original_array,
  /external/tensorflow/tensorflow/compiler/xla/
literal_util.h 1050 int64 dim2_size = values.begin()->size(); local
1075 int64 dim2_size = values.size(); local
    [all...]

Completed in 203 milliseconds