Searched defs:dim1_size (Results 1 - 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/toco/tensorflow_graph_matching/
H A Dresolve_svdf.cc182 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 Dlstm_utils.cc55 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 Dfusion_test.cc267 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 Dliteral_util.h1049 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