Searched refs:input_flat_size (Results 1 – 4 of 4) sorted by relevance
83 const int input_flat_size = GetTensorShape(op_context.input).FlatSize(); in Eval() local85 TF_LITE_ENSURE_EQ(context, input_flat_size, output_flat_size); in Eval()87 for (int i = 0; i < input_flat_size; i++) { in Eval()
205 int* input_flat_size, Shape* shape) { in ImportShape() argument232 if (input_flat_size != nullptr) *input_flat_size = 0; in ImportShape()238 if (input_flat_size == nullptr) return tensorflow::Status::OK(); in ImportShape()240 return NumElements(input_dims_only_sizes, input_flat_size); in ImportShape()342 int input_flat_size, in ImportTensorData() argument344 CHECK_GE(output_data->size(), input_flat_size); in ImportTensorData()346 if (num_elements_in_tensor == input_flat_size) { in ImportTensorData()351 input_flat_size * sizeof(T)) { in ImportTensorData()354 num_elements_in_tensor < input_flat_size) { in ImportTensorData()364 for (; i < input_flat_size; ++i) { in ImportTensorData()[all …]
108 const int input_flat_size = RequiredBufferSizeForShape(input_shape); in ExportFloatArray() local120 sizeof(*input_data) * input_flat_size)); in ExportFloatArray()130 const int input_flat_size = RequiredBufferSizeForShape(input_shape); in ExportFloatArray() local135 std::vector<float> shuffled_data(input_flat_size); in ExportFloatArray()
444 const int input_flat_size = RequiredBufferSizeForShape(input_shape); in ProcessTensorFlowReshapeOperator() local446 CHECK_GE(input_flat_size, product_non_wildcard_dims) in ProcessTensorFlowReshapeOperator()450 shape_data[wildcard_index] = input_flat_size / product_non_wildcard_dims; in ProcessTensorFlowReshapeOperator()460 CHECK_EQ(input_flat_size, RequiredBufferSizeForShape(output_shape)) in ProcessTensorFlowReshapeOperator()