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

  /external/tensorflow/tensorflow/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 96 int dim2_size, const Array& original_array,
  /external/tensorflow/tensorflow/compiler/xla/
literal_util.h 482 int64 dim2_size = values.begin()->size(); local
484 Array3D<NativeT> array(dim0_size, dim1_size, dim2_size);
493 CHECK_EQ(dim2_size, dim2);
507 int64 dim2_size = values.size(); local
510 Array4D<NativeT> array(dim0_size, dim1_size, dim2_size, dim3_size);
523 CHECK_EQ(dim2_size, dim2);

Completed in 646 milliseconds