Searched refs:input_flat_size (Results 1 – 3 of 3) sorted by relevance
200 int* input_flat_size, Shape* shape) { in ImportShape() argument227 if (input_flat_size != nullptr) *input_flat_size = 0; in ImportShape()233 if (input_flat_size == nullptr) return tensorflow::Status::OK(); in ImportShape()235 return NumElements(input_dims_only_sizes, input_flat_size); in ImportShape()325 int input_flat_size, in ImportTensorData() argument327 CHECK_GE(output_data->size(), input_flat_size); in ImportTensorData()329 if (num_elements_in_tensor == input_flat_size) { in ImportTensorData()334 input_flat_size * sizeof(T)) { in ImportTensorData()337 num_elements_in_tensor < input_flat_size) { in ImportTensorData()347 for (; i < input_flat_size; ++i) { in ImportTensorData()[all …]
105 const int input_flat_size = RequiredBufferSizeForShape(input_shape); in ExportFloatArray() local117 sizeof(*input_data) * input_flat_size)); in ExportFloatArray()127 const int input_flat_size = RequiredBufferSizeForShape(input_shape); in ExportFloatArray() local132 std::vector<float> shuffled_data(input_flat_size); in ExportFloatArray()
443 const int input_flat_size = RequiredBufferSizeForShape(input_shape); in ProcessTensorFlowReshapeOperator() local445 CHECK_GE(input_flat_size, product_non_wildcard_dims) in ProcessTensorFlowReshapeOperator()449 shape_data[wildcard_index] = input_flat_size / product_non_wildcard_dims; in ProcessTensorFlowReshapeOperator()459 CHECK_EQ(input_flat_size, RequiredBufferSizeForShape(output_shape)) in ProcessTensorFlowReshapeOperator()