Searched refs:dim2_size (Results 1 – 3 of 3) sorted by relevance
| /external/tensorflow/tensorflow/lite/toco/graph_transformations/ |
| D | lstm_utils.cc | 56 int dim2_size, const Array& original_array, in CopySubArrayToArray() argument
59 bool is_bias = dim2_size == 1; in CopySubArrayToArray()
60 Shape shape = is_bias ? Shape({dim1_size}) : Shape({dim1_size, dim2_size}); in CopySubArrayToArray()
67 start_idx1, start_idx2, buffer, dim2_size, 0, 0, dim1_size, in CopySubArrayToArray()
68 dim2_size); in CopySubArrayToArray()
|
| D | lstm_utils.h | 96 int dim2_size, const Array& original_array,
|
| /external/tensorflow/tensorflow/compiler/xla/ |
| D | literal_util.h | 449 int64 dim2_size = values.begin()->size(); in CreateR3Projected() local
451 Array3D<NativeT> array(dim0_size, dim1_size, dim2_size); in CreateR3Projected()
460 CHECK_EQ(dim2_size, dim2); in CreateR3Projected()
474 int64 dim2_size = values.size(); in CreateR4Projected() local
477 Array4D<NativeT> array(dim0_size, dim1_size, dim2_size, dim3_size); in CreateR4Projected()
490 CHECK_EQ(dim2_size, dim2); in CreateR4Projected()
|