Searched refs:dims_array (Results 1 - 2 of 2) sorted by relevance

/external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
H A Didentify_lstm.cc368 string dims_array = fc_output_split->inputs[0]; local
370 DeleteArrayIfUnused(dims_array, model);
H A Dpropagate_fixed_sizes.cc286 auto& dims_array = model->GetArray(op->inputs[0]); local
287 if (!dims_array.has_shape()) {
291 if (!dims_array.buffer) {
295 CHECK(dims_array.data_type == ArrayDataType::kInt32) << "dims must be int32";
296 CHECK_LE(RequiredBufferSizeForShape(dims_array.shape()), 4)
300 dims_array.GetBuffer<ArrayDataType::kInt32>().data;

Completed in 185 milliseconds