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

  /external/tensorflow/tensorflow/lite/toco/
import_tensorflow.cc 200 int* input_flat_size, Shape* shape) {
227 if (input_flat_size != nullptr) *input_flat_size = 0;
233 if (input_flat_size == nullptr) return tensorflow::Status::OK();
235 return NumElements(input_dims_only_sizes, input_flat_size);
325 int input_flat_size,
327 CHECK_GE(output_data->size(), input_flat_size);
329 if (num_elements_in_tensor == input_flat_size) {
334 input_flat_size * sizeof(T)) {
337 num_elements_in_tensor < input_flat_size) {
368 int input_flat_size; local
386 int input_flat_size; local
404 int input_flat_size; local
421 int input_flat_size; local
438 int input_flat_size; local
455 int input_flat_size; local
482 int input_flat_size; local
    [all...]
export_tensorflow.cc 105 const int input_flat_size = RequiredBufferSizeForShape(input_shape); local
117 sizeof(*input_data) * input_flat_size));
127 const int input_flat_size = RequiredBufferSizeForShape(input_shape); local
132 std::vector<float> shuffled_data(input_flat_size);
    [all...]
  /external/tensorflow/tensorflow/lite/toco/graph_transformations/
propagate_fixed_sizes.cc 443 const int input_flat_size = RequiredBufferSizeForShape(input_shape); local
445 CHECK_GE(input_flat_size, product_non_wildcard_dims)
449 shape_data[wildcard_index] = input_flat_size / product_non_wildcard_dims;
459 CHECK_EQ(input_flat_size, RequiredBufferSizeForShape(output_shape))
    [all...]

Completed in 79 milliseconds