Home
last modified time | relevance | path

Searched defs:input_flat_size (Results 1 – 4 of 4) sorted by relevance

/external/tensorflow/tensorflow/lite/kernels/
Dsqueeze.cc83 const int input_flat_size = GetTensorShape(op_context.input).FlatSize(); in Eval() local
/external/tensorflow/tensorflow/lite/toco/
Dimport_tensorflow.cc205 int* input_flat_size, Shape* shape) { in ImportShape()
342 int input_flat_size, in ImportTensorData()
385 int input_flat_size; in ImportFloatArray() local
403 int input_flat_size; in ImportComplex64Array() local
421 int input_flat_size; in ImportQuint8Array() local
438 int input_flat_size; in ImportInt32Array() local
455 int input_flat_size; in ImportUint32Array() local
472 int input_flat_size; in ImportInt64Array() local
489 int input_flat_size; in ImportBoolArray() local
516 int input_flat_size; in ImportStringArray() local
Dexport_tensorflow.cc108 const int input_flat_size = RequiredBufferSizeForShape(input_shape); in ExportFloatArray() local
130 const int input_flat_size = RequiredBufferSizeForShape(input_shape); in ExportFloatArray() local
/external/tensorflow/tensorflow/lite/toco/graph_transformations/
Dpropagate_fixed_sizes.cc444 const int input_flat_size = RequiredBufferSizeForShape(input_shape); in ProcessTensorFlowReshapeOperator() local