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

  /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
identify_lstm.cc 368 string dims_array = fc_output_split->inputs[0]; local
370 DeleteArrayIfUnused(dims_array, model);
propagate_fixed_sizes.cc 286 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;
    [all...]

Completed in 70 milliseconds