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

  /external/tensorflow/tensorflow/contrib/lite/kernels/
basic_rnn.cc 77 TfLiteIntArray* output_size_array = TfLiteIntArrayCreate(2); local
78 output_size_array->data[0] = batch_size;
79 output_size_array->data[1] = num_units;
81 context->ResizeTensor(context, output, output_size_array));
unidirectional_sequence_rnn.cc 82 TfLiteIntArray* output_size_array = TfLiteIntArrayCreate(3); local
83 output_size_array->data[0] = (time_major) ? max_time : batch_size;
84 output_size_array->data[1] = (time_major) ? batch_size : max_time;
85 output_size_array->data[2] = num_units;
87 context->ResizeTensor(context, output, output_size_array));
svdf.cc 100 TfLiteIntArray* output_size_array = TfLiteIntArrayCreate(2); local
101 output_size_array->data[0] = batch_size;
102 output_size_array->data[1] = num_units;
104 context->ResizeTensor(context, output, output_size_array));
fully_connected.cc 125 TfLiteIntArray* output_size_array = TfLiteIntArrayCreate(2); local
126 output_size_array->data[0] = batch_size;
127 output_size_array->data[1] = num_units;
129 context->ResizeTensor(context, output, output_size_array));
  /external/tensorflow/tensorflow/contrib/lite/toco/graph_transformations/
propagate_fixed_sizes.cc 752 const auto& output_size_array = model->GetArray(output_size_name); local
753 CHECK(output_size_array.data_type == ArrayDataType::kInt32);
754 CHECK(output_size_array.has_shape());
755 const auto& output_size_shape = output_size_array.shape();
758 if (!output_size_array.buffer) {
762 output_size_array.GetBuffer<ArrayDataType::kInt32>().data;
    [all...]

Completed in 298 milliseconds