Searched defs:dim1_size (Results 1 - 4 of 4) sorted by relevance
/external/tensorflow/tensorflow/contrib/lite/toco/tensorflow_graph_matching/ |
H A D | resolve_svdf.cc | 182 int dim1_size = 1; local 205 dim1_size *= shape.dim(d).size(); 230 new float[dim0_size * dim1_size]); 232 dim0_size, dim1_size, transposed_tensor.get()); local 234 allocated_tensor_shape->add_dim()->set_size(dim1_size);
|
/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/ |
H A D | lstm_utils.cc | 55 const string& tensor_name, int dim1_size, 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, 54 CopySubArrayToArray(Model* model, string* array_name, const string& tensor_name, int dim1_size, int dim2_size, const Array& original_array, int start_idx1, int start_idx2) argument
|
/external/tensorflow/tensorflow/compiler/xla/tests/ |
H A D | fusion_test.cc | 267 const int64 dim1_size = 1024; local 269 ShapeUtil::MakeShapeWithLayout(F32, {rand_dim0_size, dim1_size}, {1, 0}); 288 for (int j = 0; j < dim1_size; ++j) {
|
/external/tensorflow/tensorflow/compiler/xla/ |
H A D | literal_util.h | 1049 int64 dim1_size = values.size(); local 1052 Array3D<NativeT> array(dim0_size, dim1_size, dim2_size); 1064 CHECK_EQ(dim1_size, dim1); 1074 int64 dim1_size = projection_z; local 1078 Array4D<NativeT> array(dim0_size, dim1_size, dim2_size, dim3_size); 1080 for (int64 dim1 = 0; dim1 < dim1_size; ++dim1) { 1183 const int64 dim1_size = values.begin()->size(); local 1185 CHECK_EQ(dim1_size, shape().dimensions(1)); 1194 CHECK_EQ(dim1_size, dim1);
|
Completed in 177 milliseconds